Spaces:
Sleeping
Sleeping
Update requirements.txt
Browse files- Attend-and-Excite +1 -1
- model.py +2 -2
- requirements.txt +3 -2
Attend-and-Excite
CHANGED
@@ -1 +1 @@
|
|
1 |
-
Subproject commit
|
|
|
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 |
-
|
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 |
-
|
|
|
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.
|
|
|
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
|