broyang commited on
Commit
c6a69fb
·
verified ·
1 Parent(s): 8212f68

Update anime_model.py

Browse files
Files changed (1) hide show
  1. 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, base_model_id, task_name):
23
  print("Initializing base model: ", base_model_id, " with ", task_name)
24
- self.base_model_id = base_model_id
25
- self.task_name = 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 = ""