ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Removed all submodules. Submodule free now, yay.

Browse Source
This commit is contained in:
Jaret Burkett
2025-04-18 10:39:15 -06:00
parent bd2de5b74e
commit bfe29e2151
18 changed files with 1246 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
toolkit/train_tools.py
View File

@@ -6,11 +6,6 @@ import time
from typing import TYPE_CHECKING, Union, List
import sys
from torch.cuda.amp import GradScaler
from toolkit.paths import SD_SCRIPTS_ROOT
sys.path.append(SD_SCRIPTS_ROOT)
from diffusers import (
DDPMScheduler,