Julian Bilcke
commited on
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:
|