robinhad commited on
Commit
61849bb
·
verified ·
1 Parent(s): 91208a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ For more information on `huggingface_hub` Inference API support, please check th
9
  #client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
10
  lora_name = "robinhad/UAlpaca-1.1-Mistral-7B"
11
 
12
- from peft import PeftModel
13
  from transformers import LlamaTokenizer, LlamaForCausalLM, BitsAndBytesConfig
14
  from torch import bfloat16
15
  model_name = "mistralai/Mistral-7B-v0.1"
 
9
  #client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
10
  lora_name = "robinhad/UAlpaca-1.1-Mistral-7B"
11
 
12
+ from peft import PeftModel, PeftConfig
13
  from transformers import LlamaTokenizer, LlamaForCausalLM, BitsAndBytesConfig
14
  from torch import bfloat16
15
  model_name = "mistralai/Mistral-7B-v0.1"