Merge pull request #112 from if-ai/patch-1

MrForExample · web-flow · commit a64400cad56a · 2024-03-19T23:35:25.000+01:00
Update interface.py
diff --git a/crm/imagedream/ldm/interface.py b/crm/imagedream/ldm/interface.py
@@ -112,11 +112,12 @@ def register_schedule(
         self.register_buffer(
             "log_one_minus_alphas_cumprod", to_torch(np.log(1.0 - alphas_cumprod))
         )
+        eps = 1e-8  # Added this epsilon to avoid divide by zero errors
         self.register_buffer(
-            "sqrt_recip_alphas_cumprod", to_torch(np.sqrt(1.0 / alphas_cumprod))
+            "sqrt_recip_alphas_cumprod", to_torch(np.sqrt(1.0 / (alphas_cumprod + eps)))
         )
         self.register_buffer(
-            "sqrt_recipm1_alphas_cumprod", to_torch(np.sqrt(1.0 / alphas_cumprod - 1))
+            "sqrt_recipm1_alphas_cumprod", to_torch(np.sqrt(1.0 / (alphas_cumprod + eps) - 1))
         )
 
         # calculations for posterior q(x_{t-1} | x_t, x_0)

Original file line number	Diff line number	Diff line change
`@@ -112,11 +112,12 @@ def register_schedule(`
`112`	`112`	`self.register_buffer(`
`113`	`113`	`"log_one_minus_alphas_cumprod", to_torch(np.log(1.0 - alphas_cumprod))`
`114`	`114`	`)`
	`115`	`+ eps = 1e-8 # Added this epsilon to avoid divide by zero errors`
`115`	`116`	`self.register_buffer(`
`116`		`- "sqrt_recip_alphas_cumprod", to_torch(np.sqrt(1.0 / alphas_cumprod))`
	`117`	`+ "sqrt_recip_alphas_cumprod", to_torch(np.sqrt(1.0 / (alphas_cumprod + eps)))`
`117`	`118`	`)`
`118`	`119`	`self.register_buffer(`
`119`		`- "sqrt_recipm1_alphas_cumprod", to_torch(np.sqrt(1.0 / alphas_cumprod - 1))`
	`120`	`+ "sqrt_recipm1_alphas_cumprod", to_torch(np.sqrt(1.0 / (alphas_cumprod + eps) - 1))`
`120`	`121`	`)`
`121`	`122`
`122`	`123`	`# calculations for posterior q(x_{t-1} \| x_t, x_0)`