jbilcke-hf HF staff commited on
Commit
ada2e97
·
1 Parent(s): e4a3a7d

allow for deeper learning

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -575,16 +575,14 @@ class VideoTrainerUI:
575
  """Get information about the selected model type"""
576
  if model_type == "hunyuan_video":
577
  return """### HunyuanVideo (LoRA)
578
- - Best for learning complex video generation patterns
579
- - Required VRAM: ~47GB minimum
580
  - Recommended batch size: 1-2
581
  - Typical training time: 2-4 hours
582
  - Default resolution: 49x512x768
583
- - Default LoRA rank: 128"""
584
 
585
  elif model_type == "ltx_video":
586
  return """### LTX-Video (LoRA)
587
- - Lightweight video model
588
  - Required VRAM: ~18GB minimum
589
  - Recommended batch size: 1-4
590
  - Typical training time: 1-3 hours
@@ -978,13 +976,13 @@ class VideoTrainerUI:
978
  with gr.Row():
979
  lora_rank = gr.Dropdown(
980
  label="LoRA Rank",
981
- choices=["16", "32", "64", "128", "256"],
982
  value="128",
983
  type="value"
984
  )
985
  lora_alpha = gr.Dropdown(
986
  label="LoRA Alpha",
987
- choices=["16", "32", "64", "128", "256"],
988
  value="128",
989
  type="value"
990
  )
 
575
  """Get information about the selected model type"""
576
  if model_type == "hunyuan_video":
577
  return """### HunyuanVideo (LoRA)
578
+ - Required VRAM: ~48GB minimum
 
579
  - Recommended batch size: 1-2
580
  - Typical training time: 2-4 hours
581
  - Default resolution: 49x512x768
582
+ - Default LoRA rank: 128 (~600 MB)"""
583
 
584
  elif model_type == "ltx_video":
585
  return """### LTX-Video (LoRA)
 
586
  - Required VRAM: ~18GB minimum
587
  - Recommended batch size: 1-4
588
  - Typical training time: 1-3 hours
 
976
  with gr.Row():
977
  lora_rank = gr.Dropdown(
978
  label="LoRA Rank",
979
+ choices=["16", "32", "64", "128", "256", "512", "1024"],
980
  value="128",
981
  type="value"
982
  )
983
  lora_alpha = gr.Dropdown(
984
  label="LoRA Alpha",
985
+ choices=["16", "32", "64", "128", "256", "512", "1024"],
986
  value="128",
987
  type="value"
988
  )