songweig commited on
Commit
70acd79
1 Parent(s): f2e61df

update enviroment

Browse files
Files changed (3) hide show
  1. app.py +2 -1
  2. models/region_diffusion.py +1 -0
  3. requirements.txt +5 -5
app.py CHANGED
@@ -112,6 +112,7 @@ def main():
112
  # get token maps from plain text to image generation.
113
  begin_time = time.time()
114
  if model.selfattn_maps is None and model.crossattn_maps is None:
 
115
  model.register_tokenmap_hooks()
116
  else:
117
  model.reset_attention_maps()
@@ -344,7 +345,7 @@ def main():
344
  '',
345
  5,
346
  0.3,
347
- 0.3,
348
  6,
349
  0.5,
350
  None,
 
112
  # get token maps from plain text to image generation.
113
  begin_time = time.time()
114
  if model.selfattn_maps is None and model.crossattn_maps is None:
115
+ model.remove_tokenmap_hooks()
116
  model.register_tokenmap_hooks()
117
  else:
118
  model.reset_attention_maps()
 
345
  '',
346
  5,
347
  0.3,
348
+ 0.5,
349
  6,
350
  0.5,
351
  None,
models/region_diffusion.py CHANGED
@@ -41,6 +41,7 @@ class RegionDiffusion(nn.Module):
41
  self.selfattn_maps = None
42
  self.crossattn_maps = None
43
  self.color_loss = torch.nn.functional.mse_loss
 
44
  self.forward_replacement_hooks = []
45
 
46
  print(f'[INFO] loaded stable diffusion!')
 
41
  self.selfattn_maps = None
42
  self.crossattn_maps = None
43
  self.color_loss = torch.nn.functional.mse_loss
44
+ self.forward_hooks = []
45
  self.forward_replacement_hooks = []
46
 
47
  print(f'[INFO] loaded stable diffusion!')
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu113
2
- torch==1.11.0
3
- torchvision==0.12.0
4
  diffusers==0.12.1
5
- transformers==4.25.1
6
  numpy==1.24.2
7
  seaborn==0.12.2
8
- accelerate==0.18.0
9
  scikit-learn==0.24.1
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu117
2
+ torch==1.13.1
3
+ torchvision==0.14.1
4
  diffusers==0.12.1
5
+ transformers==4.26.0
6
  numpy==1.24.2
7
  seaborn==0.12.2
8
+ accelerate==0.16.0
9
  scikit-learn==0.24.1