Refactor activation function and loss calculation (#325)

2025-09-01 04:14:38 +08:00 · 2024-01-17 05:42:56 +08:00
parent ce7b65e8c4
commit 2ba5d3db14
2 changed files with 3 additions and 3 deletions
--- a/stable_diffusion/stable_diffusion/init.py
+++ b/stable_diffusion/stable_diffusion/init.py
@@ -155,5 +155,5 @@ class StableDiffusion:

    def decode(self, x_t):
        x = self.autoencoder.decode(x_t)
-        x = mx.minimum(1, mx.maximum(0, x / 2 + 0.5))
+        x = mx.clip(x / 2 + 0.5, 0, 1)
        return x