Hamid Shojanazeri 1 سال پیش
والد
کامیت
c4e96af6ee
1فایلهای تغییر یافته به همراه0 افزوده شده و 1 حذف شده
  1. 0 1
      inference/inference.py

+ 0 - 1
inference/inference.py

@@ -13,7 +13,6 @@ 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
 
 def main(
     model_name,