XiangpengYang commited on
Commit
bee795e
·
1 Parent(s): 10919c8

save weight

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -118,7 +118,12 @@ def preload_models(controller, default_model_path, default_lora_name, acc_lora_p
118
  try:
119
  print(f"[preload] Loading base model: {default_model_path}")
120
  controller.update_diffusion_transformer(default_model_path)
121
- controller.update_base_model(default_model_path)
 
 
 
 
 
122
 
123
  print(f"[preload] Loading VideoCoF LoRA: {default_lora_name}")
124
  controller.update_lora_model(default_lora_name)
 
118
  try:
119
  print(f"[preload] Loading base model: {default_model_path}")
120
  controller.update_diffusion_transformer(default_model_path)
121
+ # update_base_model expects files under Personalized_Model; skip if not present
122
+ base_candidate = os.path.join(controller.personalized_model_dir, os.path.basename(default_model_path))
123
+ if os.path.exists(base_candidate):
124
+ controller.update_base_model(os.path.basename(base_candidate))
125
+ else:
126
+ print(f"[preload] Skip update_base_model (not found at {base_candidate})")
127
 
128
  print(f"[preload] Loading VideoCoF LoRA: {default_lora_name}")
129
  controller.update_lora_model(default_lora_name)