Skip to content

Commit

Permalink
re-solve returning same output issue
Browse files Browse the repository at this point in the history
  • Loading branch information
esw0116 committed Jan 16, 2024
1 parent 1daf14f commit 76ed990
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions luciddreamer.py
Original file line number Diff line number Diff line change
Expand Up @@ -167,10 +167,6 @@ def make_controlnet_inpaint_condition(self, image, image_mask):
return image

def run(self, rgb_cond, txt_cond, neg_txt_cond, pcdgenpath, seed, diff_steps, render_camerapath, model_name=None, example_name=None):
if self.for_gradio:
# Delete gsplat.ply if exists
if os.path.exists('./gsplat.ply'):
os.remove('./gsplat.ply')
gaussians = self.create(
rgb_cond, txt_cond, neg_txt_cond, pcdgenpath, seed, diff_steps, model_name, example_name)
gallery, depth = self.render_video(render_camerapath, example_name=example_name)
Expand Down Expand Up @@ -217,6 +213,10 @@ def cleaner(self):
shutil.rmtree(dpath)
except OSError as e:
print("Error: %s - %s." % (e.filename, e.strerror))
if self.for_gradio:
# Delete gsplat.ply if exists
if os.path.exists('./gsplat.ply'):
os.remove('./gsplat.ply')

def render_video(self, preset, example_name=None, progress=gr.Progress()):
if example_name and example_name != 'DON\'T':
Expand Down

0 comments on commit 76ed990

Please sign in to comment.