update enviroment

app.py

@@ -112,6 +112,7 @@ def main():
         # get token maps from plain text to image generation.
         begin_time = time.time()
         if model.selfattn_maps is None and model.crossattn_maps is None:
+            model.remove_tokenmap_hooks()
             model.register_tokenmap_hooks()
         else:
             model.reset_attention_maps()
@@ -344,7 +345,7 @@ def main():
                     '',
                     5,
                     0.3,
-                    0.3,
+                    0.5,
                     6,
                     0.5,
                     None,

models/region_diffusion.py

@@ -41,6 +41,7 @@ class RegionDiffusion(nn.Module):
         self.selfattn_maps = None
         self.crossattn_maps = None
         self.color_loss = torch.nn.functional.mse_loss
+        self.forward_hooks = []
         self.forward_replacement_hooks = []
 
         print(f'[INFO] loaded stable diffusion!')

requirements.txt

@@ -1,9 +1,9 @@
---extra-index-url https://download.pytorch.org/whl/cu113
-torch==1.11.0
-torchvision==0.12.0
+--extra-index-url https://download.pytorch.org/whl/cu117
+torch==1.13.1
+torchvision==0.14.1
 diffusers==0.12.1
-transformers==4.25.1
+transformers==4.26.0
 numpy==1.24.2
 seaborn==0.12.2
-accelerate==0.18.0
+accelerate==0.16.0
 scikit-learn==0.24.1