Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
bee795e
1
Parent(s):
10919c8
save weight
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)
|