use torch suffix consistently
This commit is contained in:
parent
48630b3dfe
commit
5d31e85709
|
@ -1693,7 +1693,8 @@ def convert_extract_checkpoint(
|
||||||
config_file: Optional[str] = None,
|
config_file: Optional[str] = None,
|
||||||
vae_file: Optional[str] = None,
|
vae_file: Optional[str] = None,
|
||||||
) -> Tuple[bool, str]:
|
) -> Tuple[bool, str]:
|
||||||
dest_name = os.path.join(conversion.cache_path, f"{name}-torch", "working")
|
temp_name = f"{name}-torch"
|
||||||
|
dest_name = os.path.join(conversion.cache_path, temp_name, "working")
|
||||||
model_index = os.path.join(dest_name, "model_index.json")
|
model_index = os.path.join(dest_name, "model_index.json")
|
||||||
|
|
||||||
if os.path.exists(dest_name) and os.path.exists(model_index):
|
if os.path.exists(dest_name) and os.path.exists(model_index):
|
||||||
|
@ -1706,7 +1707,7 @@ def convert_extract_checkpoint(
|
||||||
)
|
)
|
||||||
if extract_checkpoint(
|
if extract_checkpoint(
|
||||||
conversion,
|
conversion,
|
||||||
name,
|
temp_name,
|
||||||
source,
|
source,
|
||||||
config_file=config_file,
|
config_file=config_file,
|
||||||
is_inpainting=is_inpainting,
|
is_inpainting=is_inpainting,
|
||||||
|
|
Loading…
Reference in New Issue