From c5947c998f284dd7a93c4a6875efaa5f4917e88e Mon Sep 17 00:00:00 2001 From: Ean Garvey <87458719+monorimet@users.noreply.github.com> Date: Mon, 11 Sep 2023 21:14:06 -0500 Subject: [PATCH] Update sd_annotation.py --- apps/stable_diffusion/src/utils/sd_annotation.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/apps/stable_diffusion/src/utils/sd_annotation.py b/apps/stable_diffusion/src/utils/sd_annotation.py index 75a6d3352d..a1d7ac8fcb 100644 --- a/apps/stable_diffusion/src/utils/sd_annotation.py +++ b/apps/stable_diffusion/src/utils/sd_annotation.py @@ -99,8 +99,7 @@ def load_lower_configs(base_model_id=None): elif version == "inpaint_v2": version = "v2_1base" - config_bucket = "gs://shark_tank/sd_tuned_configs/09112023/" - config_bucket_fallback = "gs://shark_tank/sd_tuned_configs/" + config_bucket = "gs://shark_tank/sd_tuned_configs/" device, device_spec_args = get_device_args() spec = "" @@ -161,12 +160,8 @@ def load_lower_configs(base_model_id=None): lowering_config_dir = os.path.join(WORKDIR, "configs", config_name) print("Loading lowering config file from ", lowering_config_dir) - try: - full_gs_url = config_bucket + config_name - download_public_file(full_gs_url, lowering_config_dir, True) - except: - full_gs_url = config_bucket_fallback + config_name - download_public_file(full_gs_url, lowering_config_dir, True) + full_gs_url = config_bucket + config_name + download_public_file(full_gs_url, lowering_config_dir, True) return lowering_config_dir