Update app.py

app.py

@@ -41,7 +41,7 @@ def infer(
         raise ValueError("Please specify the base model name or path")
 
     pipe = StableDiffusionPipeline.from_pretrained(model_id, 
-                                                   torch_dtype=dtype, 
+                                                   torch_dtype=torch_dtype, 
                                                    safety_checker=None).to(device)
     pipe.unet = PeftModel.from_pretrained(pipe.unet, unet_sub_dir)
     pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, text_encoder_sub_dir)