Skip to content
Snippets Groups Projects
Commit c4e96af6 authored by Hamid Shojanazeri's avatar Hamid Shojanazeri
Browse files

clean up

parent 7d2e0682
No related branches found
No related tags found
No related merge requests found
...@@ -13,7 +13,6 @@ from typing import List ...@@ -13,7 +13,6 @@ from typing import List
from transformers import LlamaTokenizer from transformers import LlamaTokenizer
from safety_utils import get_safety_checker from safety_utils import get_safety_checker
from model_utils import load_model, load_peft_model, load_llama_from_config from model_utils import load_model, load_peft_model, load_llama_from_config
from accelerate import init_empty_weights
def main( def main(
model_name, model_name,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment