Spaces:
Runtime error
Runtime error
Clean up
Browse files
model.py
CHANGED
@@ -41,7 +41,8 @@ class Model:
|
|
41 |
self.scheduler_type)
|
42 |
self.rng = random.Random()
|
43 |
|
44 |
-
|
|
|
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':
|