testing
Browse files- modeling_prismatic.py +2 -2
modeling_prismatic.py
CHANGED
@@ -429,14 +429,14 @@ class PrismaticForConditionalGeneration(PrismaticPreTrainedModel):
|
|
429 |
f"=> `past_key_values` = {past_key_values is not None}\n"
|
430 |
f"=> `use_cache` = {use_cache}"
|
431 |
)
|
432 |
-
|
433 |
# Unpack `language_model_output` and return PrismaticCausalLMOutputWithPast (or tuple if not `return_dict`)
|
434 |
if not return_dict:
|
435 |
if output_projector_features and (projected_patch_embeddings is not None):
|
436 |
return *language_model_output, projected_patch_embeddings
|
437 |
|
438 |
return language_model_output
|
439 |
-
|
440 |
return PrismaticCausalLMOutputWithPast(
|
441 |
loss=language_model_output.loss,
|
442 |
logits=language_model_output.logits,
|
|
|
429 |
f"=> `past_key_values` = {past_key_values is not None}\n"
|
430 |
f"=> `use_cache` = {use_cache}"
|
431 |
)
|
432 |
+
print('HUGGINGFACE HELLO')
|
433 |
# Unpack `language_model_output` and return PrismaticCausalLMOutputWithPast (or tuple if not `return_dict`)
|
434 |
if not return_dict:
|
435 |
if output_projector_features and (projected_patch_embeddings is not None):
|
436 |
return *language_model_output, projected_patch_embeddings
|
437 |
|
438 |
return language_model_output
|
439 |
+
|
440 |
return PrismaticCausalLMOutputWithPast(
|
441 |
loss=language_model_output.loss,
|
442 |
logits=language_model_output.logits,
|