jbilcke-hf HF Staff commited on
Commit
aeb51a1
·
1 Parent(s): bd3ee62
Files changed (1) hide show
  1. vms/ui/project/tabs/train_tab.py +1 -4
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: