login

app.py

@@ -208,6 +208,15 @@ loras = [
 ]
 
 #--------------------------------------------------Model Initialization-----------------------------------------------------------------------------------------#
+from diffusers import AutoencoderKL
+
+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)
+
 
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"