Update README.md
Browse files
README.md
CHANGED
@@ -29,10 +29,12 @@ pipe = pipeline(
|
|
29 |
"timpal0l/Mistral-7B-v0.1-flashback-v2-instruct",
|
30 |
device_map="auto"
|
31 |
)
|
|
|
32 |
text = """
|
33 |
Hur många ägg har jag? Jag hade 10 ägg, sen gav jag bort 5 ägg.
|
34 |
Sen fick jag 3 ägg av en kompis.
|
35 |
"""
|
|
|
36 |
generated = pipe(f"USER:{text}ASSISTANT:", max_length=512, temperature=0.6)
|
37 |
print(generated[0]["generated_text"].split("ASSISTANT: ")[1:][0])
|
38 |
```
|
|
|
29 |
"timpal0l/Mistral-7B-v0.1-flashback-v2-instruct",
|
30 |
device_map="auto"
|
31 |
)
|
32 |
+
|
33 |
text = """
|
34 |
Hur många ägg har jag? Jag hade 10 ägg, sen gav jag bort 5 ägg.
|
35 |
Sen fick jag 3 ägg av en kompis.
|
36 |
"""
|
37 |
+
|
38 |
generated = pipe(f"USER:{text}ASSISTANT:", max_length=512, temperature=0.6)
|
39 |
print(generated[0]["generated_text"].split("ASSISTANT: ")[1:][0])
|
40 |
```
|