1580158
/
stable-diffusion-aws-extension
mirror of https://github.com/awslabs/stable-diffusion-aws-extension.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix sync

pull/845/head
Xiujuan Li 2024-07-09 14:03:46 +08:00
parent b906f9eda2
commit 3d644e6990
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
build_scripts/comfy/comfy_proxy.py
View File

@ -538,13 +538,13 @@ if is_on_ec2:
if prepare_type in ['default', 'inputs']:
logger.info(f" sync input files")
# s5cmd_syn_input_command = f's5cmd --log=error sync --delete=true {DIR3}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/input/"'
s5cmd_syn_input_command = f's5cmd {DIR3}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/input/"'
s5cmd_syn_input_command = f'aws s3 sync --delete {DIR3}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/input/"'
logger.info(f"sync input files start {s5cmd_syn_input_command}")
os.system(s5cmd_syn_input_command)
if prepare_type in ['default', 'models']:
logger.info(f" sync models files")
# s5cmd_syn_model_command = f's5cmd --log=error sync --delete=true {DIR1}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/models/"'
s5cmd_syn_model_command = f's5cmd {DIR1}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/models/"'
s5cmd_syn_model_command = f'aws s3 sync --delete {DIR1}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/models/"'
logger.info(f"sync models files start {s5cmd_syn_model_command}")
os.system(s5cmd_syn_model_command)
logger.info(f"Files changed in:: {need_prepare} {prepare_type} {DIR2} {DIR1} {DIR3}")
@ -603,7 +603,7 @@ if is_on_ec2:
elif (str(directory).endswith(f"{DIR3}" if DIR3.startswith("/") else f"/{DIR3}")
or str(filepath) == DIR3 or str(filepath) == f'./{DIR3}' or f"{DIR3}/" in filepath):
logger.info(f" sync input files: {filepath}")
s5cmd_syn_input_command = f's5cmd --log=error sync --delete=true {DIR3}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/input/"'
s5cmd_syn_input_command = f'aws s3 sync --delete {DIR3}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/input/"'
# s5cmd_syn_input_command = f'/usr/local/bin/s5cmd sync {DIR3}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/input/"'
# 判断文件写完后再同步
@ -625,7 +625,7 @@ if is_on_ec2:
elif (str(directory).endswith(f"{DIR1}" if DIR1.startswith("/") else f"/{DIR1}")
or str(filepath) == DIR1 or str(filepath) == f'./{DIR1}' or f"{DIR1}/" in filepath):
logger.info(f" sync models files: {filepath}")
s5cmd_syn_model_command = f's5cmd --log=error sync --delete=true {DIR1}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/models/"'
s5cmd_syn_model_command = f'aws s3 sync --delete {DIR1}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/models/"'
# s5cmd_syn_model_command = f'/usr/local/bin/s5cmd sync {DIR1}/ "s3://{bucket_name}/comfy/{comfy_endpoint}/{prepare_version}/models/"'
# 判断文件写完后再同步

1
build_scripts/install_comfy.sh
View File

@ -75,6 +75,7 @@ else
/venv/bin/python3 -m pip install fastapi
/venv/bin/python3 -m pip install uvicorn
/venv/bin/python3 -m pip install watchdog
/venv/bin/python3 -m pip install awscli
/venv/bin/python3 -m pip install python-dotenv
/venv/bin/python3 -m pip install httpx
/venv/bin/python3 -s -m pip install onnxruntime