Hamid Shojanazeri 1 년 전
부모
커밋
4320528d40
1개의 변경된 파일1개의 추가작업 그리고 5개의 파일을 삭제
  1. 1 5
      inference/checkpoint_converter_fsdp_hf.py

+ 1 - 5
inference/checkpoint_converter_fsdp_hf.py

@@ -7,12 +7,8 @@ import fire
 import torch
 import os
 import sys
-from typing import List
-
 from transformers import LlamaTokenizer
-from safety_utils import get_safety_checker
-from model_utils import load_model, load_peft_model, load_llama_from_config
-from accelerate import init_empty_weights
+from model_utils import  load_llama_from_config
 # Get the current file's directory
 current_directory = os.path.dirname(os.path.abspath(__file__))