NIRVANALAN commited on
Commit
6953e16
1 Parent(s): d8e6e2b
Files changed (1) hide show
  1. nsr/train_util_diffusion.py +3 -1
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