lang-uk
/

OmniGEC-Fluency-12B

Text Generation

image-text-to-text

GrammaticalErrorCorrection

text2text-generation

text-generation-inference

Model card Files Files and versions

rkovalchuk commited on Apr 27

Commit

d982199

·

verified ·

1 Parent(s): 918a56f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -126,7 +126,7 @@ text = "She go to school every day ."
 # Choose formatting func accordingly to base model Gemma/Aya
 prompt = formatting_prompts_func(text)
-out = model.generate(**tok(prompt, return_tensors="pt"), max_new_tokens=128)
 print(tok.decode(out[0], skip_special_tokens=True))
 ```

 # Choose formatting func accordingly to base model Gemma/Aya
 prompt = formatting_prompts_func(text)
+out = model.generate(**tok(prompt, return_tensors="pt"), max_new_tokens=1600)
 print(tok.decode(out[0], skip_special_tokens=True))
 ```