Bug fixes, speed improvements, compatability adjustments withdiffusers updates

2023-09-13 07:03:53 -06:00
parent d8d1e6fd1e
commit ae70200d3c
8 changed files with 52 additions and 11 deletions
--- a/run.py
+++ b/run.py
@@ -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