ZhiliShen commited on
Commit
ccd476d
·
verified ·
1 Parent(s): 9d94139

Upload modeling_gem.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_gem.py +1 -0
modeling_gem.py CHANGED
@@ -314,6 +314,7 @@ class GEM(PreTrainedModel, GenerationMixin):
314
  show_progress_bar=show_progress_bar,
315
  convert_to_tensor=convert_to_tensor,
316
  instruction=instruction,
 
317
  return_generation=return_generation,
318
  **generation_kwargs,
319
  )
 
314
  show_progress_bar=show_progress_bar,
315
  convert_to_tensor=convert_to_tensor,
316
  instruction=instruction,
317
+ prompt=prompt,
318
  return_generation=return_generation,
319
  **generation_kwargs,
320
  )