Update app.py

app.py

@@ -57,8 +57,6 @@ def generate(
     url = '',
 ):
     if torch.cuda.is_available():
-
-        pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
         
         if not use_img2img:
             pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
@@ -85,9 +83,12 @@ def generate(
         if use_lora:
             pipe.load_lora_weights(lora)
             pipe.fuse_lora(lora_scale)
-            
+        
+        pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+        
         else:
             pipe.to(device)
+            
         
     generator = torch.Generator().manual_seed(seed)