ironjr commited on
Commit
7d1d8e0
1 Parent(s): 29e9977

Upload luciddreamer.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. luciddreamer.py +2 -2
luciddreamer.py CHANGED
@@ -213,7 +213,7 @@ class LucidDreamer:
213
  except OSError as e:
214
  print("Error: %s - %s." % (e.filename, e.strerror))
215
 
216
- def render_video(self, preset, example_name=None):
217
  if example_name and example_name != 'DON\'T':
218
  videopath = os.path.join('examples', f'{example_name}_{preset}.mp4')
219
  depthpath = os.path.join('examples', f'depth_{example_name}_{preset}.mp4')
@@ -255,7 +255,7 @@ class LucidDreamer:
255
  imageio.mimwrite(depthpath, depthlist, fps=60, quality=8)
256
  return videopath, depthpath
257
 
258
- def training(self):
259
  if not self.scene:
260
  raise('Build 3D Scene First!')
261
 
 
213
  except OSError as e:
214
  print("Error: %s - %s." % (e.filename, e.strerror))
215
 
216
+ def render_video(self, preset, example_name=None, progress=gr.Progress()):
217
  if example_name and example_name != 'DON\'T':
218
  videopath = os.path.join('examples', f'{example_name}_{preset}.mp4')
219
  depthpath = os.path.join('examples', f'depth_{example_name}_{preset}.mp4')
 
255
  imageio.mimwrite(depthpath, depthlist, fps=60, quality=8)
256
  return videopath, depthpath
257
 
258
+ def training(self, progress=gr.Progress()):
259
  if not self.scene:
260
  raise('Build 3D Scene First!')
261