Julian Bilcke
commited on
Commit
·
5c80d15
1
Parent(s):
77ce18f
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),
|