Update README.md
Browse files
README.md
CHANGED
@@ -28,10 +28,29 @@ prompt = "How to make banana bread?"
|
|
28 |
formatted_prompt = f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n"
|
29 |
|
30 |
model = TextGeneration(model_path="hf:nm-testing/TinyLlama-1.1B-Chat-v1.0-pruned50-ds")
|
31 |
-
print(model(formatted_prompt, max_new_tokens=
|
32 |
|
33 |
"""
|
|
|
34 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
|
36 |
"""
|
37 |
```
|
|
|
28 |
formatted_prompt = f"<|im_start|>user\n{prompt}<|im_end|>\n<|im_start|>assistant\n"
|
29 |
|
30 |
model = TextGeneration(model_path="hf:nm-testing/TinyLlama-1.1B-Chat-v1.0-pruned50-ds")
|
31 |
+
print(model(formatted_prompt, max_new_tokens=200).generations[0].text)
|
32 |
|
33 |
"""
|
34 |
+
Sure, here's a recipe for making banana bread:
|
35 |
|
36 |
+
Ingredients:
|
37 |
+
- 2 ripe bananas
|
38 |
+
- 1 cup all-purpose flour
|
39 |
+
- 1 cup bread flour
|
40 |
+
- 1 cup all-purpose flour
|
41 |
+
- 1 cup bread flour
|
42 |
+
- 1 cup melted coconut oil
|
43 |
+
- 1/2 cup granulated sugar
|
44 |
+
- 1/2 cup water
|
45 |
+
- 1/2 cup melted coconut oil
|
46 |
+
- 1/2 cup granulated sugar
|
47 |
+
- 1/2 cup melted coconut oil
|
48 |
+
- 1/2 cup granulated sugar
|
49 |
+
- 1/2 cup melted coconut oil
|
50 |
+
- 1/2 cup granulated sugar
|
51 |
+
- 1/2 cup melted coconut oil
|
52 |
+
- 1/2 cup granulated sugar
|
53 |
+
- 1/2 cup
|
54 |
|
55 |
"""
|
56 |
```
|