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]
|