Spaces:
Running
on
Zero
Running
on
Zero
NIRVANALAN
commited on
Commit
•
6953e16
1
Parent(s):
d8e6e2b
update
Browse files
nsr/train_util_diffusion.py
CHANGED
@@ -205,7 +205,8 @@ class TrainLoopDiffusionWithRec(TrainLoop):
|
|
205 |
rec_model(latent=ddpm_latent,
|
206 |
behaviour='decode_after_vae_no_render'))
|
207 |
|
208 |
-
if export_mesh:
|
|
|
209 |
import mcubes
|
210 |
import trimesh
|
211 |
dump_path = f'{logger.get_dir()}/mesh/'
|
@@ -373,6 +374,7 @@ class TrainLoopDiffusionWithRec(TrainLoop):
|
|
373 |
print('logged video to: ', f'{vid_dump_path}')
|
374 |
|
375 |
del vis, pred_vis, micro, pred,
|
|
|
376 |
|
377 |
return vid_dump_path, mesh_dump_path
|
378 |
|
|
|
205 |
rec_model(latent=ddpm_latent,
|
206 |
behaviour='decode_after_vae_no_render'))
|
207 |
|
208 |
+
# if export_mesh:
|
209 |
+
if False:
|
210 |
import mcubes
|
211 |
import trimesh
|
212 |
dump_path = f'{logger.get_dir()}/mesh/'
|
|
|
374 |
print('logged video to: ', f'{vid_dump_path}')
|
375 |
|
376 |
del vis, pred_vis, micro, pred,
|
377 |
+
mesh_dump_path = './logs/LSGM/inference/Objaverse/i23d/dit-L2/gradio_app/mesh/cfg=4_sample-0.obj'
|
378 |
|
379 |
return vid_dump_path, mesh_dump_path
|
380 |
|