hysts HF staff commited on
Commit
44dd855
1 Parent(s): e3c9822

Update requirements.txt

Browse files
Files changed (3) hide show
  1. Attend-and-Excite +1 -1
  2. model.py +2 -2
  3. requirements.txt +3 -2
Attend-and-Excite CHANGED
@@ -1 +1 @@
1
- Subproject commit 1b67cfc19cd3952e390dbb8047ccd126471567f2
 
1
+ Subproject commit 41620338367f980b9d73752360ffd2557d8ddf5d
model.py CHANGED
@@ -27,8 +27,8 @@ class Model:
27
  def load_model(self, model_id: str) -> None:
28
  if model_id == self.model_id:
29
  return
30
- self.model = AttendAndExcitePipeline.from_pretrained(
31
- model_id, revision='249dd2d').to(self.device)
32
  self.tokenizer = self.model.tokenizer
33
  self.model_id = model_id
34
 
 
27
  def load_model(self, model_id: str) -> None:
28
  if model_id == self.model_id:
29
  return
30
+ self.model = AttendAndExcitePipeline.from_pretrained(model_id).to(
31
+ self.device)
32
  self.tokenizer = self.model.tokenizer
33
  self.model_id = model_id
34
 
requirements.txt CHANGED
@@ -1,7 +1,8 @@
1
- diffusers==0.3.0
 
2
  ftfy==6.1.1
3
  jupyter
4
  opencv-python-headless==4.7.0.68
5
  pyrallis==0.3.1
6
  torch==1.13.1
7
- transformers==4.23.1
 
1
+ accelerate==0.19.0
2
+ diffusers==0.12.1
3
  ftfy==6.1.1
4
  jupyter
5
  opencv-python-headless==4.7.0.68
6
  pyrallis==0.3.1
7
  torch==1.13.1
8
+ transformers==4.29.2