Hugging Face's logo

Spaces:
Coots
/
ArqonzChat
Runtime error

App Files Community
Coots commited on
Commit
936a720
·
verified ·
1 Parent(s): 1ddc0ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from pathlib import Path
6
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
7
 
8
  # Model: Falcon-rw-1b (decoder-only, not instruction-tuned)
9
- model_id = "tiiuae/falcon-rw-1b"
10
  tokenizer = AutoTokenizer.from_pretrained(model_id)
11
  model = AutoModelForCausalLM.from_pretrained(model_id)
12
 
 
6
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
7
 
8
  # Model: Falcon-rw-1b (decoder-only, not instruction-tuned)
9
+ model_id = "TheBloke/Mistral-7B-Instruct-v0.2-GGUF"
10
  tokenizer = AutoTokenizer.from_pretrained(model_id)
11
  model = AutoModelForCausalLM.from_pretrained(model_id)
12