Add Mac OS support for Apple Silicon (#770)
* Made an install script and auto updates env for mac * GPU sensors and initial training working for MAC. Still WIP. * Switch dataloader to single threaded until I can work around some mac pickeling issues. * Get quantization working on mac * Fix mac exclusive imports so they don't break other builds. * Add mac instructions to the UI
This commit is contained in:
1 parent
bc47fd6755
commit
171535833a
25 files changed
+1072
-71
No files matched your search
@@ -72,10 +72,7 @@ import hashlib
|
||||
|
||||
from toolkit.util.blended_blur_noise import get_blended_blur_noise
|
||||
from toolkit.util.get_model import get_model_class
|
||||
|
||||
def flush():
|
||||
torch.cuda.empty_cache()
|
||||
gc.collect()
|
||||
from toolkit.basic import flush
|
||||
|
||||
|
||||
class BaseSDTrainProcess(BaseTrainProcess):
|
||||
|
||||
Reference in new issue
Block a user