Hugging Face's logo

Spaces:
habulaj
/
squaadai
Sleeping

App Files Community
1
forplaytvplus commited on
Commit
9785c61
·
verified ·
1 Parent(s): 5b8ee78

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -115,7 +115,7 @@ def generate(
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("cuda")
119
  generator = torch.Generator().manual_seed(seed)
120
 
121
  if not use_negative_prompt:
 
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
 
121
  if not use_negative_prompt: