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:
Jaret Burkett authored and GitHub committed 2026-03-30 09:37:47 -06:00
1 parent bc47fd6755
commit 171535833a
25 files changed
+1072 -71

No files matched your search

+1 -4
View File
@@ -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):