Tech-Meld commited on
Commit
f25f274
1 Parent(s): b25bdff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def load_and_merge_lora(base_model_id, lora_id, lora_adapter_name):
18
  torch_dtype=torch.float16,
19
  variant="fp16",
20
  use_safetensors=True,
21
- ).to("cuda")
22
 
23
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(
24
  pipe.scheduler.config
 
18
  torch_dtype=torch.float16,
19
  variant="fp16",
20
  use_safetensors=True,
21
+ ).to("cpu")
22
 
23
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(
24
  pipe.scheduler.config