Hugging Face's logo

Spaces:
ASSERT-KTH
/
demos
Sleeping

App Files Community
GGmorello commited on
Commit
633c781
·
verified ·
1 Parent(s): 3875a78

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")
9
  model = LlamaForCausalLM.from_pretrained(
10
- "GGmorello/FLAMES_100k",
11
  quantization_config=BitsAndBytesConfig(
12
  load_in_4bit=True,
13
  bnb_4bit_quant_type="nf4",
 
7
 
8
  # config = PeftConfig.from_pretrained("GGmorello/FLAMES")
9
  model = LlamaForCausalLM.from_pretrained(
10
+ "GGmorello/FLAMES-100k",
11
  quantization_config=BitsAndBytesConfig(
12
  load_in_4bit=True,
13
  bnb_4bit_quant_type="nf4",