Update app.py

app.py

@@ -46,7 +46,7 @@ def infer(
     pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir)
     pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir)
 
-    if dtype in (torch.float16, torch.bfloat16):
+    if torch_dtype in (torch.float16, torch.bfloat16):
         pipe.unet.half()
         pipe.text_encoder.half()