rockylynnstein commited on
Commit
aec9cca
·
verified ·
1 Parent(s): 22f2164

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -6
README.md CHANGED
@@ -52,13 +52,16 @@ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
52
  ```python
53
  from vllm import LLM, SamplingParams
54
 
55
- llm = LLM(model="TevunahAi/granite-20b-code-instruct-8k-FP8")
56
- sampling_params = SamplingParams(temperature=0.7, max_tokens=256)
57
-
58
- prompts = ["Write a Python function to calculate fibonacci numbers:"]
59
- outputs = llm.generate(prompts, sampling_params)
 
 
 
 
60
  ```
61
-
62
  ## Quantization Details
63
 
64
  - **Target Layers:** All Linear layers except lm_head
 
52
  ```python
53
  from vllm import LLM, SamplingParams
54
 
55
+ if __name__ == '__main__':
56
+ llm = LLM(model="TevunahAi/granite-20b-code-instruct-8k-FP8")
57
+ sampling_params = SamplingParams(temperature=0.7, max_tokens=256)
58
+
59
+ outputs = llm.generate(["Write a Python fibonacci function:"], sampling_params)
60
+
61
+ for output in outputs:
62
+ print(output.outputs[0].text)
63
+ print("---DONE---")
64
  ```
 
65
  ## Quantization Details
66
 
67
  - **Target Layers:** All Linear layers except lm_head