pszemraj commited on
Commit
9066742
·
verified ·
1 Parent(s): 2238f64

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -2
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 = "What color is tuesday?"
38
- res = pipe(prompt, max_new_tokens=96, top_k=4, penalty_alpha=0.6)
 
 
 
 
 
 
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