Carlosito16 commited on
Commit
8b5df99
·
1 Parent(s): e106a6d

remove the device_map to omit the accelerate error

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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={ "device_map": "auto",
 
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