Update README.md
Browse files
README.md
CHANGED
|
@@ -28,7 +28,7 @@ model = AutoModelForCausalLM.from_pretrained("JuIm/ProGemma") tokenizer = AutoTo
|
|
| 28 |
|
| 29 |
progemma = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
| 30 |
|
| 31 |
-
sequence = progemma("
|
| 32 |
|
| 33 |
print(sequence)
|
| 34 |
|
|
|
|
| 28 |
|
| 29 |
progemma = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
| 30 |
|
| 31 |
+
sequence = progemma("\<bos>", top_k=950, max_length=100, num_return_sequences=1, do_sample=True, repetition_penalty=1.2, eos_token_id=21, pad_token_id=22, bos_token_id=20)
|
| 32 |
|
| 33 |
print(sequence)
|
| 34 |
|