rrnoa's picture
111
5f1b538
raw
history blame
481 Bytes
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