willxxy commited on
Commit
11a71a9
·
verified ·
1 Parent(s): 2dbc77d
Files changed (1) hide show
  1. 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,