Hugging Face's logo

Spaces:
rphrp1985
/
zerogpu-2
Running on Zero

App Files Community
rphrp1985 commited on
Commit
07d42bf
·
verified ·
1 Parent(s): c33ed79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,12 +27,12 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
27
 
28
  # model_id = "mistralai/Mistral-7B-v0.3"
29
 
30
- model_id = "CohereForAI/c4ai-command-r-plus-4bit"
31
 
32
 
33
  tokenizer = AutoTokenizer.from_pretrained(
34
  # model_id
35
- 'CohereForAI/c4ai-command-r-plus'
36
  , token= token,)
37
 
38
  model = AutoModelForCausalLM.from_pretrained(model_id, token= token,
 
27
 
28
  # model_id = "mistralai/Mistral-7B-v0.3"
29
 
30
+ model_id = "CohereForAI/c4ai-command-r-v01"
31
 
32
 
33
  tokenizer = AutoTokenizer.from_pretrained(
34
  # model_id
35
+ model_id
36
  , token= token,)
37
 
38
  model = AutoModelForCausalLM.from_pretrained(model_id, token= token,