saanvi-bot commited on
Commit
660c406
·
verified ·
1 Parent(s): 60842f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -8
app.py CHANGED
@@ -1,12 +1,5 @@
1
- from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
2
- from peft import prepare_model_for_kbit_training
3
- from peft import LoraConfig, get_peft_model
4
- from datasets import load_dataset
5
- import transformers
6
-
7
- # load model from hub
8
  from peft import PeftModel, PeftConfig
9
- from transformers import AutoModelForCausalLM
10
 
11
  model_name = "TheBloke/Mistral-7B-Instruct-v0.2-GPTQ"
12
  model = AutoModelForCausalLM.from_pretrained(model_name,
 
1
+ from transformers import AutoModelForCausalLM, AutoTokenizer
 
 
 
 
 
 
2
  from peft import PeftModel, PeftConfig
 
3
 
4
  model_name = "TheBloke/Mistral-7B-Instruct-v0.2-GPTQ"
5
  model = AutoModelForCausalLM.from_pretrained(model_name,