Commit
·
ea54aa8
1
Parent(s):
1c47fec
Updated generation script2
Browse files- evaluate.py +1 -1
evaluate.py
CHANGED
|
@@ -6,5 +6,5 @@ model = GPT2LMHeadModel.from_pretrained("flax-community/swe-gpt-wiki")
|
|
| 6 |
|
| 7 |
generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
|
| 8 |
set_seed(42)
|
| 9 |
-
result = generator("
|
| 10 |
print(result)
|
|
|
|
| 6 |
|
| 7 |
generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
|
| 8 |
set_seed(42)
|
| 9 |
+
result = generator("En dag för länge sedan så fanns det", max_length=150, num_return_sequences=5)
|
| 10 |
print(result)
|