mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-01-26 16:39:47 +00:00
36 lines
804 B
JSON
36 lines
804 B
JSON
{
|
|
"job": "train",
|
|
"config": {
|
|
"name": "name_of_your_model",
|
|
"base_model": "/path/to/base/model",
|
|
"training_folder": "/path/to/output/folder",
|
|
"is_v2": false,
|
|
"device": "cuda",
|
|
"gradient_accumulation_steps": 1,
|
|
"mixed_precision": "fp16",
|
|
"logging_dir": "/path/to/tensorboard/log/folder",
|
|
|
|
"process": [
|
|
{
|
|
"type": "fine_tune"
|
|
}
|
|
]
|
|
},
|
|
"meta": {
|
|
"name": "[name]",
|
|
"description": "A short description of your model",
|
|
"trigger_words": [
|
|
"put",
|
|
"trigger",
|
|
"words",
|
|
"here"
|
|
],
|
|
"version": "0.1",
|
|
"creator": {
|
|
"name": "Your Name",
|
|
"email": "your@email.com",
|
|
"website": "https://yourwebsite.com"
|
|
},
|
|
"any": "All meta data above is arbitrary, it can be whatever you want."
|
|
}
|
|
} |