File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
apps/stable_diffusion/src/utils Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments