Spaces:
Running
on
Zero
Running
on
Zero
Specify SD v1.4 model revision to avoid dependency error
Browse files
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 |
-
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(
|
31 |
+
model_id, revision='249dd2d').to(self.device)
|
32 |
self.tokenizer = self.model.tokenizer
|
33 |
self.model_id = model_id
|
34 |
|