new

app.py

@@ -283,9 +283,9 @@ state_dict = torch.load(model_ckpt_path, map_location='cpu')['state_dict']
 state_dict = {k[14:]: v for k, v in state_dict.items() if k.startswith('lrm_generator.')}
 model.load_state_dict(state_dict, strict=True)
 
-model = model.to(device1)
+model = model.to(device)
 if IS_FLEXICUBES:
-    model.init_flexicubes_geometry(device1, fovy=30.0)
+    model.init_flexicubes_geometry(device, fovy=30.0)
 model = model.eval()
 
 print('Loading Finished!')