vsrinivas commited on
Commit
d7c09eb
·
verified ·
1 Parent(s): 2c98ceb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ pipeline = transformers.pipeline(
37
  trust_remote_code=True,
38
  device_map="auto",
39
  offload_folder="off_load",
40
- offload_state_dict = True,
41
  )
42
 
43
  # def format_chat_prompt(message, chat_history):
 
37
  trust_remote_code=True,
38
  device_map="auto",
39
  offload_folder="off_load",
40
+ # offload_state_dict = True,
41
  )
42
 
43
  # def format_chat_prompt(message, chat_history):