Update app.py

app.py

@@ -79,6 +79,8 @@ def generate_image(
     seed:int=-1,
     nums:int=1):
 
+    pipe.to(device="cuda")
+
     if seed == -1:
         seed = random.randint(0, MAX_SEED)
     seed = int(seed)
@@ -117,7 +119,6 @@ def gen(
     progress=gr.Progress(track_tqdm=True)
 ):
     enable_lora(lora_scale, lora_in, lora_add)
-    pipe.to(device="cuda", dtype=torch.bfloat16)
     return generate_image(prompt,width,height,scales,steps,seed,nums)