jbilcke-hf HF Staff commited on
Commit
3822049
·
1 Parent(s): 72e9451

fix for "latest"

Browse files
Files changed (1) hide show
  1. vms/ui/project/tabs/train_tab.py +4 -2
vms/ui/project/tabs/train_tab.py CHANGED
@@ -612,11 +612,13 @@ class TrainTab(BaseTab):
612
  has_checkpoints = len(checkpoints) > 0
613
  resume_from = resume_from_checkpoint # Use the passed parameter
614
 
615
- if resume_from == "latest" and checkpoints:
616
  # Find the latest checkpoint
617
  latest_checkpoint = max(checkpoints, key=os.path.getmtime)
618
  resume_from = str(latest_checkpoint)
619
- logger.info(f"Found checkpoint at {resume_from}, will resume training")
 
 
620
 
621
  # Convert model_type display name to internal name
622
  model_internal_type = MODEL_TYPES.get(model_type)
 
612
  has_checkpoints = len(checkpoints) > 0
613
  resume_from = resume_from_checkpoint # Use the passed parameter
614
 
615
+ if resume_from and checkpoints:
616
  # Find the latest checkpoint
617
  latest_checkpoint = max(checkpoints, key=os.path.getmtime)
618
  resume_from = str(latest_checkpoint)
619
+
620
+ logger.info(f"Found checkpoint at {resume_from}, note from @julian: right now let's just resume training at 'latest'")
621
+ result_from = "latest"
622
 
623
  # Convert model_type display name to internal name
624
  model_internal_type = MODEL_TYPES.get(model_type)