Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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):
|