hysts HF staff commited on
Commit
4598779
1 Parent(s): 950d117

Specify SD v1.4 model revision to avoid dependency error

Browse files
Files changed (1) hide show
  1. model.py +2 -2
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(model_id).to(
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