Simplify
Browse files
README.md
CHANGED
|
@@ -443,12 +443,7 @@ tokenizer = T5Tokenizer.from_pretrained('jbochi/madlad400-7b-mt')
|
|
| 443 |
|
| 444 |
text = "<2it> I love pizza!"
|
| 445 |
input_ids = tokenizer(text, return_tensors="pt").input_ids
|
| 446 |
-
|
| 447 |
-
outputs = model.generate(
|
| 448 |
-
input_ids=input_ids,
|
| 449 |
-
generation_config=GenerationConfig(
|
| 450 |
-
decoder_start_token_id=0,
|
| 451 |
-
))
|
| 452 |
|
| 453 |
tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 454 |
# Adoro la pizza!
|
|
|
|
| 443 |
|
| 444 |
text = "<2it> I love pizza!"
|
| 445 |
input_ids = tokenizer(text, return_tensors="pt").input_ids
|
| 446 |
+
outputs = model.generate(input_ids=input_ids)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 447 |
|
| 448 |
tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 449 |
# Adoro la pizza!
|