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 train path

pull/852/head
Jingyi 2024-07-18 11:24:40 +08:00
parent 6efc292c55
commit 55d9a258fc
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
middleware_api/trainings/create_training_job.py
View File

@ -100,14 +100,13 @@ def _trigger_sagemaker_training_job(
train_job_name (str): training job name
"""
site_packages_s3_path = (f"aws-gcr-solutions-{region}/"
site_packages_s3_path = (f"s3://aws-gcr-solutions-{region}/"
f"stable-diffusion-aws-extension-github-mainline/{esd_version}/train.tar")
data = {
"id": train_job.id,
"training_id": train_job.id,
"sagemaker_program": "extensions/sd-webui-sagemaker/sagemaker_entrypoint_json.py",
"site_packages_s3_path": site_packages_s3_path,
"params": train_job.params,
"s3-input-path": train_job.input_s3_location,
"s3-output-path": ckpt_output_path,