Spaces:
Paused
Paused
Commit
·
8b5df99
1
Parent(s):
e106a6d
remove the device_map to omit the accelerate error
Browse files
app.py
CHANGED
@@ -100,7 +100,8 @@ def load_llm_model():
|
|
100 |
#this one is for running with GPT
|
101 |
llm = HuggingFacePipeline.from_model_id(model_id= 'lmsys/fastchat-t5-3b-v1.0',
|
102 |
task= 'text2text-generation',
|
103 |
-
model_kwargs={
|
|
|
104 |
"max_length": 256, "temperature": 0,
|
105 |
"repetition_penalty": 1.5,
|
106 |
"quantization_config": bitsandbyte_config}) #add this quantization config
|
|
|
100 |
#this one is for running with GPT
|
101 |
llm = HuggingFacePipeline.from_model_id(model_id= 'lmsys/fastchat-t5-3b-v1.0',
|
102 |
task= 'text2text-generation',
|
103 |
+
model_kwargs={
|
104 |
+
# "device_map": "auto",
|
105 |
"max_length": 256, "temperature": 0,
|
106 |
"repetition_penalty": 1.5,
|
107 |
"quantization_config": bitsandbyte_config}) #add this quantization config
|