apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-26 17:11:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

add lora ft cmd to readme

Browse source
This commit is contained in:
51616 2024-12-20 15:09:16 +00:00
parent bc7b588775
commit f2606dc29d
3 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
hyperlora/model_loading.py
View file

@ -101,7 +101,6 @@ def get_model(
device="cuda:0",
dtype=torch.bfloat16,
):
# TODO: we could use lower precision during finetuning
model_init_kwargs = dict(
pretrained_model_name_or_path=model_name_or_path,
device_map=device,