[Bugfix] [CI] Fix Tensorizer LoRA test (#20760)

Signed-off-by: Sanger Steel <sangersteel@gmail.com>
This commit is contained in:
Sanger Steel
2025-07-10 15:07:06 -04:00
committed by GitHub
parent c66e38ea4c
commit 5e53c89a74
3 changed files with 14 additions and 19 deletions

View File

@@ -102,7 +102,7 @@ class PEFTHelper:
tensorizer_config = TensorizerConfig(**tensorizer_config_dict)
tensorizer_args = tensorizer_config._construct_tensorizer_args()
from tensorizer.stream_io import open_stream
lora_config_path = os.path.join(tensorizer_config.lora_dir,
lora_config_path = os.path.join(tensorizer_config.tensorizer_dir,
"adapter_config.json")
with open_stream(lora_config_path,
mode="rb",
@@ -110,7 +110,7 @@ class PEFTHelper:
config = json.load(f)
logger.info("Successfully deserialized LoRA config from %s",
tensorizer_config.lora_dir)
tensorizer_config.tensorizer_dir)
else:
with open(lora_config_path) as f: