Spaces:
Running
Running
add XL variant mappings to _VARIANT_DIR
Browse files- train_engine.py +3 -0
train_engine.py
CHANGED
|
@@ -399,8 +399,11 @@ def save_lora_adapter(model, output_dir: str) -> None:
|
|
| 399 |
|
| 400 |
_VARIANT_DIR = {
|
| 401 |
"turbo": "acestep-v15-turbo",
|
|
|
|
| 402 |
"base": "acestep-v15-base",
|
|
|
|
| 403 |
"sft": "acestep-v15-sft",
|
|
|
|
| 404 |
}
|
| 405 |
|
| 406 |
|
|
|
|
| 399 |
|
| 400 |
_VARIANT_DIR = {
|
| 401 |
"turbo": "acestep-v15-turbo",
|
| 402 |
+
"xl-turbo": "acestep-v15-xl-turbo",
|
| 403 |
"base": "acestep-v15-base",
|
| 404 |
+
"xl-base": "acestep-v15-xl-base",
|
| 405 |
"sft": "acestep-v15-sft",
|
| 406 |
+
"xl-sft": "acestep-v15-xl-sft",
|
| 407 |
}
|
| 408 |
|
| 409 |
|