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

formatting

Browse source
This commit is contained in:
51616 2024-12-20 10:18:30 +00:00
parent e0e010301e
commit a38779abd7
6 changed files with 68 additions and 71 deletions
Download patch file Download diff file Expand all files Collapse all files

7
hyperlora/model_loading.py
View file

@ -2,13 +2,10 @@ import logging
import os
import torch
from peft import PeftModel, LoraConfig, VeraConfig, PeftConfig
from peft import LoraConfig, PeftConfig, PeftModel, VeraConfig
from peft import get_peft_config as _get_peft_config
from peft.utils import PeftType
from transformers import AutoModelForCausalLM, AutoTokenizer, AutoModel
from hyper_llm_modulator.utils.pooling import get_pooling_fn
from hyper_llm_modulator.utils.preprocessing import add_full_stop, apply_sfr_template
from transformers import AutoModel, AutoModelForCausalLM, AutoTokenizer
logger = logging.getLogger()