FlameF0X commited on
Commit
a7e01dd
·
verified ·
1 Parent(s): 64dfa83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
5
 
6
  # Initialize your small CoT / Reasoning model
7
  # (Using a pipeline; will be moved to GPU inside the decorated function)
8
- MODEL_ID = "DeepSeek-AI/DeepSeek-R1-Distill-Qwen-1.5B"
9
  generator = pipeline("text-generation", model=MODEL_ID, torch_dtype=torch.bfloat16)
10
 
11
  @spaces.GPU(duration=60) # Requests ZeroGPU allocation for the duration of the tree search
 
5
 
6
  # Initialize your small CoT / Reasoning model
7
  # (Using a pipeline; will be moved to GPU inside the decorated function)
8
+ MODEL_ID = "LiquidAI/LFM2.5-1.2B-Thinking"
9
  generator = pipeline("text-generation", model=MODEL_ID, torch_dtype=torch.bfloat16)
10
 
11
  @spaces.GPU(duration=60) # Requests ZeroGPU allocation for the duration of the tree search