Initial commit

app.py

@@ -120,7 +120,10 @@ class FaceLiftPipeline:
             workspace_dir / "checkpoints/gslrm/ckpt_0000000000021125.pt",
             map_location="cpu"
         )
-        self.gs_lrm_model.load_state_dict(checkpoint["model"])
+        # Filter out loss_calculator weights (training-only, not needed for inference)
+        state_dict = {k: v for k, v in checkpoint["model"].items() 
+                      if not k.startswith("loss_calculator.")}
+        self.gs_lrm_model.load_state_dict(state_dict)
         # Keep on CPU initially - will move to GPU in decorated function
         
         self.color_prompt_embedding = torch.load(