Upload folder using huggingface_hub
Browse files
README.md
CHANGED
|
@@ -40,7 +40,7 @@ from wedlm import LLM, SamplingParams
|
|
| 40 |
llm = LLM(model="tencent/WeDLM-7B")
|
| 41 |
|
| 42 |
prompt = "The theory of relativity states that"
|
| 43 |
-
outputs = llm.generate([prompt], SamplingParams(temperature=0.
|
| 44 |
|
| 45 |
print(outputs[0]["text"])
|
| 46 |
```
|
|
|
|
| 40 |
llm = LLM(model="tencent/WeDLM-7B")
|
| 41 |
|
| 42 |
prompt = "The theory of relativity states that"
|
| 43 |
+
outputs = llm.generate([prompt], SamplingParams(temperature=0.2, max_tokens=256))
|
| 44 |
|
| 45 |
print(outputs[0]["text"])
|
| 46 |
```
|