ksort commited on
Commit
99eeb28
1 Parent(s): 7271da2

Add new model

Browse files
Files changed (1) hide show
  1. model/model_manager.py +1 -1
model/model_manager.py CHANGED
@@ -27,7 +27,7 @@ class ModelManager:
27
  @spaces.GPU(duration=120)
28
  def generate_image_ig(self, prompt, model_name):
29
  pipe = self.load_model_pipe(model_name)
30
- if 'Stable-cascade' not in name:
31
  result = pipe(prompt=prompt).images[0]
32
  else:
33
  prior, decoder = pipe
 
27
  @spaces.GPU(duration=120)
28
  def generate_image_ig(self, prompt, model_name):
29
  pipe = self.load_model_pipe(model_name)
30
+ if 'Stable-cascade' not in model_name:
31
  result = pipe(prompt=prompt).images[0]
32
  else:
33
  prior, decoder = pipe