correções

app.py

@@ -244,6 +244,10 @@ loras = [
     # add new
 ]
 
+# Configuração inicial
+dtype = torch.float16  # Pode ser torch.bfloat16 ou torch.float32, dependendo do hardware
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
 # Initialize the base model
 base_model = "SG161222/Verus_Vision_1.0b"
 
@@ -253,7 +257,7 @@ good_vae = AutoencoderKL.from_pretrained(base_model, filename="ae.safetensors",
 # Configurar o pipeline principal do Verus Vision
 pipe = DiffusionPipeline.from_pretrained(
     base_model,
-    filename="VerusVision_1.0b_Transformer_fp16.safetensors",  # Ajuste para FP16 ou FP8
+    filename="VerusVision_1.0b_Transformer_fp16.safetensors",  # Ajuste para FP16 ou FP8 conforme necessário
     torch_dtype=dtype
 ).to(device)