robinhad commited on
Commit
be048a4
1 Parent(s): be676b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ if is_available():
11
  )
12
  else:
13
  options = {
14
- "dtype": bfloat16
15
  #"low_cpu_mem_usage": True,
16
  #"offload_state_dict": True,
17
  #"offload_folder": "offload",
 
11
  )
12
  else:
13
  options = {
14
+ "torch_dtype": bfloat16
15
  #"low_cpu_mem_usage": True,
16
  #"offload_state_dict": True,
17
  #"offload_folder": "offload",