add cpu

app.py

@@ -60,7 +60,7 @@ def get_lora_sd_pipeline(
 
     pipe = StableDiffusionPipeline.from_pretrained(
         base_model_name_or_path, torch_dtype=dtype
-    ).to(device)
+    )  # .to(device)
     pipe.unet = PeftModel.from_pretrained(
         pipe.unet, unet_sub_dir, adapter_name=adapter_name
     )
@@ -101,7 +101,7 @@ def infer(
         # добавляем lora
         pipe = get_lora_sd_pipeline(
             os.path.join(CKPT_DIR, ""), adapter_name="sd-14-lora", dtype=torch_dtype
-        ).to(device)
+        )  # .to(device)
         pipe.safety_checker = None
         print(f"LoRA adapter loaded: {pipe.unet.active_adapters}")
 
@@ -112,7 +112,7 @@ def infer(
             requires_safety_checker=False,
             safety_checker=None,
         )
-        pipe = pipe.to(device)
+        pipe = pipe  # .to(device)
 
     image = pipe(
         prompt=prompt,