Kingrane commited on
Commit
13a2bfa
·
verified ·
1 Parent(s): 4664dfb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -3,8 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
 
5
  model_name = "mistralai/Devstral-Small-2505"
6
-
7
- tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=False)
8
  model = AutoModelForCausalLM.from_pretrained(
9
  model_name,
10
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
 
3
  import torch
4
 
5
  model_name = "mistralai/Devstral-Small-2505"
6
+ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1", use_fast=False)
 
7
  model = AutoModelForCausalLM.from_pretrained(
8
  model_name,
9
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32