Spaces:
Sleeping
Sleeping
Update anime_model.py
Browse files- anime_model.py +3 -3
anime_model.py
CHANGED
|
@@ -19,10 +19,10 @@ from settings import MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES
|
|
| 19 |
class Model:
|
| 20 |
MODEL_ID = "lllyasviel/Annotators"
|
| 21 |
|
| 22 |
-
def __init__(self
|
| 23 |
print("Initializing base model: ", base_model_id, " with ", task_name)
|
| 24 |
-
self.base_model_id =
|
| 25 |
-
self.task_name =
|
| 26 |
self.pipe = self.load_pipe(base_model_id, task_name)
|
| 27 |
self.model = None
|
| 28 |
self.name = ""
|
|
|
|
| 19 |
class Model:
|
| 20 |
MODEL_ID = "lllyasviel/Annotators"
|
| 21 |
|
| 22 |
+
def __init__(self):
|
| 23 |
print("Initializing base model: ", base_model_id, " with ", task_name)
|
| 24 |
+
self.base_model_id = "nyxia/AAM-AnyLoRA-Anime-Mix"
|
| 25 |
+
self.task_name = "NormalBae"
|
| 26 |
self.pipe = self.load_pipe(base_model_id, task_name)
|
| 27 |
self.model = None
|
| 28 |
self.name = ""
|