MAISAAI
/

gemma-2b-coder

Text Generation

Generated from Trainer

text-generation-inference

Model card Files Files and versions

mrm8488 commited on Feb 22, 2024

Commit

e6af35e

·

verified ·

1 Parent(s): 322cbad

Update README.md

Files changed (1) hide show

README.md +0 -1

README.md CHANGED Viewed

@@ -120,7 +120,6 @@ def generate(
             attention_mask=attention_mask,
             generation_config=generation_config,
             return_dict_in_generate=True,
-            #output_scores=True,
             max_new_tokens=max_new_tokens,
             early_stopping=True
         )

             attention_mask=attention_mask,
             generation_config=generation_config,
             return_dict_in_generate=True,
             max_new_tokens=max_new_tokens,
             early_stopping=True
         )