pyakhurel commited on
Commit
0c99222
1 Parent(s): de7173e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,8 @@ import gradio as gr
3
  from peft import PeftModel
4
  from transformers import AutoModelForCausalLM, AutoTokenizer
5
  import transformers
 
 
6
 
7
 
8
  adapters_name = "1littlecoder/mistral-7b-mj-finetuned"
 
3
  from peft import PeftModel
4
  from transformers import AutoModelForCausalLM, AutoTokenizer
5
  import transformers
6
+ import bitsandbytes
7
+ import accelerate
8
 
9
 
10
  adapters_name = "1littlecoder/mistral-7b-mj-finetuned"