Bug fixes, speed improvements, compatability adjustments withdiffusers updates

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

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