GGmorello commited on
Commit
885566b
·
verified ·
1 Parent(s): e3edf18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import LlamaForCausalLM, AutoTokenizer, BitsAndBytesConfig
7
 
8
  config = PeftConfig.from_pretrained("GGmorello/FLAMES-20k")
9
  model = LlamaForCausalLM.from_pretrained(
10
- config.base_model_name_or_path,
11
  quantization_config=BitsAndBytesConfig(
12
  load_in_4bit=True,
13
  bnb_4bit_quant_type="nf4",
 
7
 
8
  config = PeftConfig.from_pretrained("GGmorello/FLAMES-20k")
9
  model = LlamaForCausalLM.from_pretrained(
10
+ 'codellama/codellama-7b-hf',
11
  quantization_config=BitsAndBytesConfig(
12
  load_in_4bit=True,
13
  bnb_4bit_quant_type="nf4",