forplaytvplus commited on
Commit
7c5d546
·
verified ·
1 Parent(s): 9785c61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -114,7 +114,8 @@ def generate(
114
  else:
115
  adapter_name2 = cached_loras[lora_key2]
116
  pipe.set_adapters([adapter_name1, adapter_name2], adapter_weights=[lora_scale, lora_scale2])
117
-
 
118
  pipe.to(device)
119
  generator = torch.Generator().manual_seed(seed)
120
 
 
114
  else:
115
  adapter_name2 = cached_loras[lora_key2]
116
  pipe.set_adapters([adapter_name1, adapter_name2], adapter_weights=[lora_scale, lora_scale2])
117
+
118
+ enable_model_cpu_offload()
119
  pipe.to(device)
120
  generator = torch.Generator().manual_seed(seed)
121