Skip to content

Commit b63fb36

Browse files
authored
Use path.join for the winograd config directory (huggingface#1065)
1 parent 4e92304 commit b63fb36

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

apps/stable_diffusion/src/utils/sd_annotation.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def load_winograd_configs():
7070
config_bucket = "gs://shark_tank/sd_tuned/configs/"
7171
config_name = f"{args.annotation_model}_winograd_{device}.json"
7272
full_gs_url = config_bucket + config_name
73-
winograd_config_dir = f"{WORKDIR}configs/" + config_name
73+
winograd_config_dir = os.path.join(WORKDIR, "configs", config_name)
7474
print("Loading Winograd config file from ", winograd_config_dir)
7575
download_public_file(full_gs_url, winograd_config_dir, True)
7676
return winograd_config_dir
@@ -113,7 +113,7 @@ def load_lower_configs():
113113
config_name = f"{args.annotation_model}_{version}_{args.precision}_{device}_{spec}.json"
114114

115115
full_gs_url = config_bucket + config_name
116-
lowering_config_dir = f"{WORKDIR}configs/" + config_name
116+
lowering_config_dir = os.path.join(WORKDIR, "configs", config_name)
117117
print("Loading lowering config file from ", lowering_config_dir)
118118
download_public_file(full_gs_url, lowering_config_dir, True)
119119
return lowering_config_dir

0 commit comments

Comments
 (0)