Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,10 +9,10 @@ hf_token = os.environ.get("HF_TOKEN")
|
|
9 |
model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
10 |
|
11 |
# Load the tokenizer
|
12 |
-
tokenizer = AutoTokenizer.from_pretrained(model_id,
|
13 |
|
14 |
# Load the model
|
15 |
-
model = AutoModelForCausalLM.from_pretrained(model_id,
|
16 |
|
17 |
def format_prompt(message, history):
|
18 |
prompt = "<s>"
|
|
|
9 |
model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
|
10 |
|
11 |
# Load the tokenizer
|
12 |
+
tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token)
|
13 |
|
14 |
# Load the model
|
15 |
+
model = AutoModelForCausalLM.from_pretrained(model_id, token=hf_token, device_map="auto")
|
16 |
|
17 |
def format_prompt(message, history):
|
18 |
prompt = "<s>"
|