FractalAIR commited on
Commit
e9de7ce
·
verified ·
1 Parent(s): 8b34e77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,7 +7,8 @@ import re
7
  import uuid
8
 
9
  # Load model and tokenizer
10
- our_model_path = "FractalAIResearch/Fathom-R1-14B"
 
11
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
12
 
13
  our_model = AutoModelForCausalLM.from_pretrained(our_model_path, device_map="auto", torch_dtype="auto")
 
7
  import uuid
8
 
9
  # Load model and tokenizer
10
+ our_model_path = "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B"
11
+ #our_model_path = "FractalAIResearch/Fathom-R1-14B"
12
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
13
 
14
  our_model = AutoModelForCausalLM.from_pretrained(our_model_path, device_map="auto", torch_dtype="auto")