hysts HF staff commited on
Commit
80b7378
1 Parent(s): eab3b8e
Files changed (1) hide show
  1. model.py +2 -1
model.py CHANGED
@@ -41,7 +41,8 @@ class Model:
41
  self.scheduler_type)
42
  self.rng = random.Random()
43
 
44
- def _load_pipeline(self, model_name: str,
 
45
  scheduler_type: str) -> DiffusionPipeline:
46
  repo_id = f'hysts/diffusers-anime-faces-{model_name}'
47
  if scheduler_type == 'DDPM':
 
41
  self.scheduler_type)
42
  self.rng = random.Random()
43
 
44
+ @staticmethod
45
+ def _load_pipeline(model_name: str,
46
  scheduler_type: str) -> DiffusionPipeline:
47
  repo_id = f'hysts/diffusers-anime-faces-{model_name}'
48
  if scheduler_type == 'DDPM':