Spaces:
Running
on
Zero
Running
on
Zero
NIRVANALAN
commited on
Commit
•
96737f5
1
Parent(s):
076d511
update
Browse files
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]
|