ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-26 17:29:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bug fixes, speed improvements, compatability adjustments withdiffusers updates

Browse Source
This commit is contained in:
Jaret Burkett
2023-09-13 07:03:53 -06:00
parent d8d1e6fd1e
commit ae70200d3c
8 changed files with 52 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
run.py
View File

@@ -4,7 +4,7 @@ from typing import Union, OrderedDict
sys.path.insert(0, os.getcwd())
# must come before ANY torch or fastai imports
import toolkit.cuda_malloc
# import toolkit.cuda_malloc
import argparse
from toolkit.job import get_job