Update README.md
Browse files
README.md
CHANGED
@@ -34,8 +34,14 @@ pipe = pipeline(
|
|
34 |
model="BEE-spoke-data/tFINE-900m-e16-d32-instruct",
|
35 |
# device_map="auto", # uncomment if have GPU/accelerate
|
36 |
)
|
37 |
-
prompt = "
|
38 |
-
res = pipe(
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
print(res[0]["generated_text"])
|
40 |
```
|
41 |
|
|
|
34 |
model="BEE-spoke-data/tFINE-900m-e16-d32-instruct",
|
35 |
# device_map="auto", # uncomment if have GPU/accelerate
|
36 |
)
|
37 |
+
prompt = "Write me a python script that demonstrates an advanced sorting algorithm"
|
38 |
+
res = pipe(
|
39 |
+
prompt,
|
40 |
+
max_new_tokens=384,
|
41 |
+
num_beams=4,
|
42 |
+
early_stopping=True,
|
43 |
+
no_repeat_ngram_size=6,
|
44 |
+
)
|
45 |
print(res[0]["generated_text"])
|
46 |
```
|
47 |
|