Spaces:
Paused
Paused
Update llm.py
Browse files
llm.py
CHANGED
|
@@ -23,7 +23,8 @@ model = HuggingFacePipeline.from_model_id(
|
|
| 23 |
model_id="microsoft/Phi-3-small-8k-instruct",
|
| 24 |
task="text-generation",
|
| 25 |
pipeline_kwargs={"temperature":1, "max_length":1000,
|
| 26 |
-
"repetition_penalty":1.25, "max_new_tokens": 2000,
|
|
|
|
| 27 |
)
|
| 28 |
print("loading model done!")
|
| 29 |
|
|
|
|
| 23 |
model_id="microsoft/Phi-3-small-8k-instruct",
|
| 24 |
task="text-generation",
|
| 25 |
pipeline_kwargs={"temperature":1, "max_length":1000,
|
| 26 |
+
"repetition_penalty":1.25, "max_new_tokens": 2000},
|
| 27 |
+
trust_remote_code=True
|
| 28 |
)
|
| 29 |
print("loading model done!")
|
| 30 |
|