aaa

app.py

@@ -210,15 +210,18 @@ loras = [
 #--------------------------------------------------Model Initialization-----------------------------------------------------------------------------------------#
 from diffusers import AutoencoderKL
 
+# Definir dtype antes de usar
+dtype = torch.bfloat16
+
 good_vae = AutoencoderKL.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     subfolder="vae",
     torch_dtype=dtype,
-    use_auth_token=True  # Inclui o token de autenticação
-).to(device)
+    use_auth_token=True
+).to("cuda" if torch.cuda.is_available() else "cpu")
+
 
 
-dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 base_model = "black-forest-labs/FLUX.1-dev"