add device

app.py

@@ -4,7 +4,6 @@ import random
 import gradio as gr
 import numpy as np
 import torch
-
 # import spaces #[uncomment to use ZeroGPU]
 from diffusers import StableDiffusionPipeline
 from peft import LoraConfig, PeftModel
@@ -60,7 +59,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 +100,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 +111,7 @@ def infer(
             requires_safety_checker=False,
             safety_checker=None,
         )
-        pipe = pipe  # .to(device)
+        pipe = pipe.to(device)
 
     image = pipe(
         prompt=prompt,