Minor fix for ZeroGPU (#2)

- Minor fix for ZeroGPU (4a015bec4086a00db45448189c2a4494e124a2fd)


Co-authored-by: hysts <hysts@users.noreply.huggingface.co>

app.py

@@ -55,6 +55,7 @@ pipe = load_evo_ukiyoe(device)
 pipe.scheduler = EulerDiscreteScheduler.from_config(
     pipe.scheduler.config, use_karras_sigmas=True,
 )
+pipe.to(device=device, dtype=torch.float16)
 # pipe.unet.to(memory_format=torch.channels_last)
 # pipe.vae.to(memory_format=torch.channels_last)
 # # Compile the UNet and VAE.
@@ -77,7 +78,6 @@ def generate(
     randomize_seed: bool = False,
     progress=gr.Progress(track_tqdm=True),
 ):
-    pipe.to(device)
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
 
@@ -143,4 +143,4 @@ with gr.Blocks(css=css) as demo:
                 Sakana AIは、本モデルの使用によって生じた直接的または間接的な損失に対して、結果に関わらず、一切の責任を負いません。
                 利用者は、本モデルの使用に伴うリスクを十分に理解し、自身の判断で使用することが必要です。""")
 
-demo.queue().launch()
+demo.queue().launch()