robinhad commited on
Commit
f6701c9
1 Parent(s): 3a96ce3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ else:
12
  options = {
13
  "low_cpu_mem_usage": True,
14
  "offload_state_dict": True,
15
- "offload_dir": "offload",
16
  "device_map": "auto",
17
  }
18
 
 
12
  options = {
13
  "low_cpu_mem_usage": True,
14
  "offload_state_dict": True,
15
+ "offload_folder": "offload",
16
  "device_map": "auto",
17
  }
18