Spaces:
Running
Running
Commit
·
aeb51a1
1
Parent(s):
bd3ee62
fix
Browse files
vms/ui/project/tabs/train_tab.py
CHANGED
@@ -555,8 +555,7 @@ For image-to-video tasks, 'index' (usually with index 0) is most common as it co
|
|
555 |
return self.handle_training_start(
|
556 |
model_type, model_version, training_type,
|
557 |
lora_rank, lora_alpha, train_steps, batch_size, learning_rate,
|
558 |
-
save_iterations, repo_id, progress,
|
559 |
-
pretrained_lora_path=str(latest_lora_dir)
|
560 |
)
|
561 |
|
562 |
def connect_events(self) -> None:
|
@@ -862,7 +861,6 @@ For image-to-video tasks, 'index' (usually with index 0) is most common as it co
|
|
862 |
save_iterations, repo_id,
|
863 |
progress=gr.Progress(),
|
864 |
resume_from_checkpoint=None,
|
865 |
-
pretrained_lora_path=None,
|
866 |
):
|
867 |
"""Handle training start with proper log parser reset and checkpoint detection"""
|
868 |
|
@@ -931,7 +929,6 @@ For image-to-video tasks, 'index' (usually with index 0) is most common as it co
|
|
931 |
precomputation_items=precomputation_items,
|
932 |
lr_warmup_steps=lr_warmup_steps,
|
933 |
progress=progress,
|
934 |
-
pretrained_lora_path=pretrained_lora_path,
|
935 |
custom_prompt_prefix=custom_prompt_prefix
|
936 |
)
|
937 |
except Exception as e:
|
|
|
555 |
return self.handle_training_start(
|
556 |
model_type, model_version, training_type,
|
557 |
lora_rank, lora_alpha, train_steps, batch_size, learning_rate,
|
558 |
+
save_iterations, repo_id, progress,
|
|
|
559 |
)
|
560 |
|
561 |
def connect_events(self) -> None:
|
|
|
861 |
save_iterations, repo_id,
|
862 |
progress=gr.Progress(),
|
863 |
resume_from_checkpoint=None,
|
|
|
864 |
):
|
865 |
"""Handle training start with proper log parser reset and checkpoint detection"""
|
866 |
|
|
|
929 |
precomputation_items=precomputation_items,
|
930 |
lr_warmup_steps=lr_warmup_steps,
|
931 |
progress=progress,
|
|
|
932 |
custom_prompt_prefix=custom_prompt_prefix
|
933 |
)
|
934 |
except Exception as e:
|