minhdang commited on
Commit
711fb55
·
verified ·
1 Parent(s): 8666022

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -12,14 +12,11 @@ Join us : 🌟TeamTonic🌟 is always making cool demos! Join our active builder
12
  """
13
 
14
  bnb_config = BitsAndBytesConfig(
15
- load_in_4bit=True,
16
- bnb_4bit_quant_type="nf4",
17
- bnb_4bit_use_double_quant=True,
18
- bnb_4bit_compute_dtype=torch.bfloat16
19
  )
20
 
21
 
22
- model_id = "CohereForAI/c4ai-command-r-v01"
23
  tokenizer = AutoTokenizer.from_pretrained(model_id)
24
  model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config)
25
 
 
12
  """
13
 
14
  bnb_config = BitsAndBytesConfig(
15
+ load_in_8bit=True,
 
 
 
16
  )
17
 
18
 
19
+ model_id = "refuelai/Llama-3-Refueled"
20
  tokenizer = AutoTokenizer.from_pretrained(model_id)
21
  model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config)
22