Spaces:
Running
Running
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=
|
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),
|