Hugging Face's logo

Spaces:
HFHAB
/
mixtral-46.7b-fastapi
Runtime error

App Files Community
HFHAB commited on
Commit
d82c856
·
verified ·
1 Parent(s): eaea91c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
7
 
8
  app = FastAPI()
9
 
10
- model_id = "mistralai/Mistral-7B-v0.1
11
  model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
12
  tokenizer = AutoTokenizer.from_pretrained(model_id)
13
 
 
7
 
8
  app = FastAPI()
9
 
10
+ model_id = "mistralai/Mistral-7B-v0.1"
11
  model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
12
  tokenizer = AutoTokenizer.from_pretrained(model_id)
13