From eb33ee524827a5805460268c53bce384e29196b9 Mon Sep 17 00:00:00 2001
From: Extraltodeus <extraltodeus@gmail.com>
Date: Fri, 24 May 2024 17:31:20 +0200
Subject: [PATCH] Update experimental_temperature.py

---
 experimental_temperature.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/experimental_temperature.py b/experimental_temperature.py
index ffed7bb..d7f3c43 100644
--- a/experimental_temperature.py
+++ b/experimental_temperature.py
@@ -8,8 +8,9 @@ import types
 import os
 
 def exists(val):
-    return val is not None       
+    return val is not None
 
+# better than a division by 0 hey
 abs_mean = lambda x: torch.where(torch.isnan(x) | torch.isinf(x), torch.zeros_like(x), x).abs().mean()
 
 class temperature_patcher():
@@ -204,4 +205,4 @@ class CLIPTemperaturePatchDual:
                 clip_encoder_instance_g = m.cond_stage_model.clip_g.transformer.text_model.encoder
                 clip_encoder_instance_g.forward = types.MethodType(new_forward, clip_encoder_instance_g)
         
-        return (m,)
\ No newline at end of file
+        return (m,)