NIRVANALAN commited on
Commit
96737f5
1 Parent(s): 076d511
Files changed (1) hide show
  1. nsr/lsgm/flow_matching_trainer.py +2 -2
nsr/lsgm/flow_matching_trainer.py CHANGED
@@ -693,9 +693,9 @@ class FlowMatchingEngine(TrainLoop3DDiffusionLSGM_crossattn):
693
  **kwargs,
694
  ):
695
 
696
- output_model, output_video = './logs/LSGM/inference/Objaverse/i23d/dit-L2/gradio_app/mesh/cfg=4.0_sample-0.ply', './logs/LSGM/inference/Objaverse/i23d/dit-L2/gradio_app/triplane_cfg=4.0_sample-0.mp4'
697
 
698
- return output_model, output_video
699
 
700
  camera = th.load('assets/objv_eval_pose.pt', map_location=dist_util.dev())[:]
701
  inp_img = th.from_numpy(inp_img).permute(2,0,1).unsqueeze(0) / 127.5 - 1 # to [-1,1]
 
693
  **kwargs,
694
  ):
695
 
696
+ # output_model, output_video = './logs/LSGM/inference/Objaverse/i23d/dit-L2/gradio_app/mesh/cfg=4.0_sample-0.ply', './logs/LSGM/inference/Objaverse/i23d/dit-L2/gradio_app/triplane_cfg=4.0_sample-0.mp4'
697
 
698
+ # return output_model, output_video
699
 
700
  camera = th.load('assets/objv_eval_pose.pt', map_location=dist_util.dev())[:]
701
  inp_img = th.from_numpy(inp_img).permute(2,0,1).unsqueeze(0) / 127.5 - 1 # to [-1,1]