from diffusers import StableDiffusionDepth2ImgPipeline
import torch

# Carga del modelo
print("[DEBUG] Iniciando prueba de carga del modelo Marigold...")
try:
    pipe = StableDiffusionDepth2ImgPipeline.from_pretrained(
        "prs-eth/marigold-depth-v1-0",
        torch_dtype=torch.float16
    )
    pipe.to("cuda")
    print("[DEBUG] Modelo Marigold cargado exitosamente.")
except Exception as e:
    print(f"[ERROR] Error al cargar el modelo: {e}")
    raise e