mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
2.1.826
This commit is contained in:
@@ -5,10 +5,9 @@ import sys
|
||||
print('[System ARGV] ' + str(sys.argv))
|
||||
|
||||
root = os.path.dirname(os.path.abspath(__file__))
|
||||
backend_path = os.path.join(root, 'backend', 'headless')
|
||||
sys.path += [root, backend_path]
|
||||
|
||||
sys.path.append(root)
|
||||
os.chdir(root)
|
||||
|
||||
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
|
||||
os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0.0"
|
||||
os.environ["GRADIO_SERVER_PORT"] = "7865"
|
||||
@@ -88,19 +87,19 @@ def download_models():
|
||||
return
|
||||
|
||||
|
||||
def ini_fcbh_args():
|
||||
def ini_args():
|
||||
from args_manager import args
|
||||
return args
|
||||
|
||||
|
||||
prepare_environment()
|
||||
build_launcher()
|
||||
args = ini_fcbh_args()
|
||||
args = ini_args()
|
||||
|
||||
|
||||
if args.cuda_device is not None:
|
||||
os.environ['CUDA_VISIBLE_DEVICES'] = str(args.cuda_device)
|
||||
print("Set device to:", args.cuda_device)
|
||||
if args.gpu_device_id is not None:
|
||||
os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu_device_id)
|
||||
print("Set device to:", args.gpu_device_id)
|
||||
|
||||
|
||||
download_models()
|
||||
|
||||
Reference in New Issue
Block a user