vinesmsuic commited on
Commit
0ddac36
·
1 Parent(s): d43fe7c
Files changed (2) hide show
  1. model/model_manager.py +2 -2
  2. requirements.txt +2 -2
model/model_manager.py CHANGED
@@ -81,10 +81,10 @@ class ModelManager:
81
  self.enable_nsfw = enable_nsfw
82
  self.load_guard(enable_nsfw)
83
  self.loaded_models = {}
84
- if do_pre_download:
85
- pre_download_all_models(include_video=False)
86
  if do_debug_packages:
87
  debug_packages()
 
 
88
 
89
  try:
90
  self.generate_image_ig.zerogpu = True
 
81
  self.enable_nsfw = enable_nsfw
82
  self.load_guard(enable_nsfw)
83
  self.loaded_models = {}
 
 
84
  if do_debug_packages:
85
  debug_packages()
86
+ if do_pre_download:
87
+ pre_download_all_models(include_video=False)
88
 
89
  try:
90
  self.generate_image_ig.zerogpu = True
requirements.txt CHANGED
@@ -7,7 +7,7 @@ h5py
7
  xformers
8
  numpy>=1.23.5
9
  pandas<2.0.0
10
- peft
11
  torch
12
  torchvision
13
  torchaudio
@@ -25,7 +25,7 @@ pillow~=9.5.0
25
  einops>=0.4.1
26
  torch-fidelity>=0.3.0
27
  setuptools>=59.5.0
28
- transformers
29
  torchmetrics>=0.6.0
30
  kornia>=0.6
31
  diffusers>=0.18.0
 
7
  xformers
8
  numpy>=1.23.5
9
  pandas<2.0.0
10
+ peft~=0.10.0
11
  torch
12
  torchvision
13
  torchaudio
 
25
  einops>=0.4.1
26
  torch-fidelity>=0.3.0
27
  setuptools>=59.5.0
28
+ transformers~=4.37.2
29
  torchmetrics>=0.6.0
30
  kornia>=0.6
31
  diffusers>=0.18.0