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':
|