ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-01-08 03:17:47 +00:00
commit 18113b2248
1 changed files with 13 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

26
unsloth/models/sentence_transformer.py
View file

@ -42,10 +42,10 @@ import shutil
def _save_pretrained_torchao(
self,
save_directory,
tokenizer=None,
torchao_config=None,
push_to_hub=False,
token=None,
tokenizer = None,
torchao_config = None,
push_to_hub = False,
token = None,
):
self.save_pretrained(save_directory)
@ -71,10 +71,10 @@ def _save_pretrained_torchao(
break
except:
pass
transformer_dir = os.path.join(save_directory, transformer_path)
transformer_dir = os.path.abspath(transformer_dir)
if tokenizer is None:
tokenizer = self.tokenizer
@ -97,27 +97,27 @@ def _save_pretrained_torchao(
unsloth_save_pretrained_torchao(
inner_model,
transformer_dir,
tokenizer=tokenizer,
torchao_config=torchao_config,
push_to_hub=push_to_hub,
token=token,
tokenizer = tokenizer,
torchao_config = torchao_config,
push_to_hub = push_to_hub,
token = token,
)
# avoid `0_Transformer-torchao`, it was either this or fix modules.json
torchao_dir = transformer_dir + "-torchao"
if os.path.exists(torchao_dir):
if not os.path.exists(transformer_dir):
os.makedirs(transformer_dir, exist_ok=True)
os.makedirs(transformer_dir, exist_ok = True)
# move contents
for item in os.listdir(torchao_dir):
s = os.path.join(torchao_dir, item)
d = os.path.join(transformer_dir, item)
if os.path.isdir(s):
shutil.copytree(s, d, dirs_exist_ok=True)
shutil.copytree(s, d, dirs_exist_ok = True)
else:
shutil.copy2(s, d)
# remove torchao dir
shutil.rmtree(torchao_dir)