rhaymison
/

gemma-portuguese-luana-2b

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

rhaymison commited on Mar 25

Commit

6afda49

•

1 Parent(s): a3a3d27

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -81,7 +81,7 @@ device = "cuda:0"
 inputs = tokenizer2(text, return_tensors="pt").to(device)
-outputs = model2.generate(**inputs, max_new_tokens=100, do_sample=False)
 output = tokenizer2.decode(outputs[0], skip_special_tokens=True, skip_prompt=True)
 print(output.replace("model"," "))

 inputs = tokenizer2(text, return_tensors="pt").to(device)
+outputs = model2.generate(**inputs, max_new_tokens=256, do_sample=False)
 output = tokenizer2.decode(outputs[0], skip_special_tokens=True, skip_prompt=True)
 print(output.replace("model"," "))