Spaces:

ndhieunguyen
/

Lang2mol-Diff

Running

ndhieunguyen commited on Jun 6, 2024

Commit

dfe8f6f

•

1 Parent(s): 9c86aa3

fix: oor

Files changed (1) hide show

src/improved_diffusion/gaussian_diffusion.py CHANGED Viewed

@@ -694,7 +694,7 @@ class GaussianDiffusion:
                 )
                 yield out
                 img = out["sample"]
-            my_bar.progress(pro + 1, text="Processing")
         my_bar.empty()
     def p_sample_loop_langevin_progressive(

                 )
                 yield out
                 img = out["sample"]
+            my_bar.progress(pro, text="Processing")
         my_bar.empty()
     def p_sample_loop_langevin_progressive(