Spaces:

123123aa123
/

UniGeo

Running on Zero

123123aa123 commited on 12 days ago

Commit

2893a92

verified ·

1 Parent(s): fdf498d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -98,7 +98,7 @@ def load_models():
     if vggt_model is None:
         print("Loading VGGT...")
-        vggt_model = VGGT.from_pretrained(VGGT_PATH, torch_dtype=dtype).to(device).eval()
     if wan_pipe is None:
         print("Loading Wan...")

     if vggt_model is None:
         print("Loading VGGT...")
+        vggt_model = VGGT.from_pretrained(VGGT_PATH).to(device).eval()
     if wan_pipe is None:
         print("Loading Wan...")