Add .sh run script for macOS & linux, fix error on macs with low vram. (#737)

* Add .sh run script

* Update extract_feature_print.py

* Remove `requirements_macOS.txt`
This commit is contained in:
Naozumi
2023-07-13 15:05:35 +08:00
committed by GitHub
parent 5b9d9b045a
commit 86ed98aaca
2 changed files with 46 additions and 0 deletions

View File

@@ -1,6 +1,7 @@
import os, sys, traceback
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0.0"
# device=sys.argv[1]
n_part = int(sys.argv[2])