ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 19:21:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Removed train config, updating it, and added my llvae as pytorch model

Browse Source
This commit is contained in:
Jaret Burkett
2023-07-20 14:12:28 -06:00
parent c29b9d075f
commit 982e0be7a9
2 changed files with 118 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
config/examples/train.example.json
View File

@@ -1,36 +0,0 @@
{
"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."
}
}