Hugging Face's logo

Spaces:
minhdang
/
Command-R
Runtime error

App Files Community
minhdang commited on
Commit
a8636d7
·
verified ·
1 Parent(s): f7f4304

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ nf4_config = BitsAndBytesConfig(
13
  bnb_4bit_compute_dtype=torch.bfloat16
14
  )
15
 
16
- model_id = "AetherResearch/Cerebrum-1.0-8x7b"
17
  tokenizer = AutoTokenizer.from_pretrained(model_id)
18
  model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=nf4_config)
19
 
 
13
  bnb_4bit_compute_dtype=torch.bfloat16
14
  )
15
 
16
+ model_id = "CohereForAI/c4ai-command-r-v01"
17
  tokenizer = AutoTokenizer.from_pretrained(model_id)
18
  model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=nf4_config)
19