jbilcke-hf HF Staff commited on
Commit
6e3431d
·
1 Parent(s): ed18efe

debugging automatic job recovery

Browse files
vms/ui/app_ui.py CHANGED
@@ -434,14 +434,12 @@ class AppUI:
434
  training_preset = list(TRAINING_PRESETS.keys())[0]
435
  logger.warning(f"Invalid training preset '{training_preset}', using default: {training_preset}")
436
 
437
- # Rest of the function remains unchanged
438
  lora_rank_val = ui_state.get("lora_rank", DEFAULT_LORA_RANK_STR)
439
  lora_alpha_val = ui_state.get("lora_alpha", DEFAULT_LORA_ALPHA_STR)
440
  batch_size_val = int(ui_state.get("batch_size", DEFAULT_BATCH_SIZE))
441
  learning_rate_val = float(ui_state.get("learning_rate", DEFAULT_LEARNING_RATE))
442
  save_iterations_val = int(ui_state.get("save_iterations", DEFAULT_SAVE_CHECKPOINT_EVERY_N_STEPS))
443
 
444
- # Update for new UI components
445
  num_gpus_val = int(ui_state.get("num_gpus", DEFAULT_NUM_GPUS))
446
 
447
  # Calculate recommended precomputation items based on video count
 
434
  training_preset = list(TRAINING_PRESETS.keys())[0]
435
  logger.warning(f"Invalid training preset '{training_preset}', using default: {training_preset}")
436
 
 
437
  lora_rank_val = ui_state.get("lora_rank", DEFAULT_LORA_RANK_STR)
438
  lora_alpha_val = ui_state.get("lora_alpha", DEFAULT_LORA_ALPHA_STR)
439
  batch_size_val = int(ui_state.get("batch_size", DEFAULT_BATCH_SIZE))
440
  learning_rate_val = float(ui_state.get("learning_rate", DEFAULT_LEARNING_RATE))
441
  save_iterations_val = int(ui_state.get("save_iterations", DEFAULT_SAVE_CHECKPOINT_EVERY_N_STEPS))
442
 
 
443
  num_gpus_val = int(ui_state.get("num_gpus", DEFAULT_NUM_GPUS))
444
 
445
  # Calculate recommended precomputation items based on video count
vms/ui/project/services/previewing.py CHANGED
@@ -520,7 +520,6 @@ class PreviewingService:
520
  log_fn("Starting video generation...")
521
  start_time.record()
522
 
523
- # Fix for Issue #2: The pipe() expected list rather than float
524
  # Make sure negative_prompt is a list or None
525
  neg_prompt = [negative_prompt] if negative_prompt else None
526
 
 
520
  log_fn("Starting video generation...")
521
  start_time.record()
522
 
 
523
  # Make sure negative_prompt is a list or None
524
  neg_prompt = [negative_prompt] if negative_prompt else None
525
 
vms/ui/project/services/training.py CHANGED
@@ -1151,7 +1151,7 @@ class TrainingService:
1151
  # Use the internal model_type for the actual training
1152
  # But keep model_type_display for the UI
1153
  result = self.start_training(
1154
- model_type=model_internal_type,
1155
  lora_rank=params.get('lora_rank', DEFAULT_LORA_RANK_STR),
1156
  lora_alpha=params.get('lora_alpha', DEFAULT_LORA_ALPHA_STR),
1157
  train_size=params.get('train_steps', DEFAULT_NB_TRAINING_STEPS),
 
1151
  # Use the internal model_type for the actual training
1152
  # But keep model_type_display for the UI
1153
  result = self.start_training(
1154
+ model_type=model_type_internal,
1155
  lora_rank=params.get('lora_rank', DEFAULT_LORA_RANK_STR),
1156
  lora_alpha=params.get('lora_alpha', DEFAULT_LORA_ALPHA_STR),
1157
  train_size=params.get('train_steps', DEFAULT_NB_TRAINING_STEPS),