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