Update app.py

app.py

@@ -98,7 +98,7 @@ def load_models():
 
     if vggt_model is None:
         print("Loading VGGT...")
-        vggt_model = VGGT.from_pretrained(VGGT_PATH).to(device).eval()
+        vggt_model = VGGT.from_pretrained(VGGT_PATH, torch_dtype=dtype).to(device).eval()
 
     if wan_pipe is None:
         print("Loading Wan...")
@@ -133,10 +133,10 @@ def load_models():
         wan_pipe.load_lora(wan_pipe.dit, state_dict=lora_sd, alpha=1)
         wan_pipe.dit.load_state_dict(adapter_sd, strict=False)
 
-        wan_pipe.to(device)
-        wan_pipe.to(dtype=torch.bfloat16)
-
+        #wan_pipe.to(device)
+        #wan_pipe.to(dtype=torch.bfloat16)
 
+load_models()
 # =========================
 # Renderer
 # =========================
@@ -268,8 +268,6 @@ def build_estimate_rel(x, y, z, phi, theta):
 @spaces.GPU
 def infer(image, prompt, seed):
 
-    load_models()
-
 
     img = image.convert("RGB")