Spaces:
Runtime error
Runtime error
Julian Bilcke
commited on
Commit
·
ada2e97
1
Parent(s):
e4a3a7d
allow for deeper learning
Browse files
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 |
-
-
|
| 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 |
)
|