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

added parameter `site_packages_s3_path`

pull/852/head
Jingyi 2024-07-17 12:57:25 +08:00
parent 3d68a1b0d8
commit 42cd62b2e5
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
middleware_api/trainings/create_training_job.py
View File

@ -100,10 +100,14 @@ def _trigger_sagemaker_training_job(
train_job_name (str): training job name
"""
site_packages_s3_path = (f"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,
@ -138,8 +142,7 @@ def _trigger_sagemaker_training_job(
},
job_id=train_job.id,
environment={
"SITE_PACKAGES_S3_PATH": f"aws-gcr-solutions-{region}/"
f"stable-diffusion-aws-extension-github-mainline/{esd_version}/train.tar"
"SITE_PACKAGES_S3_PATH": site_packages_s3_path
}
)
est.fit(wait=False)