Hugging Face's logo

Spaces:
asadshahab
/
token-attention-visualizer
Running

App Files Community
asadshahab commited on
Commit
ba66953
·
verified ·
1 Parent(s): 6ffe751

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -4,7 +4,7 @@ from typing import Optional
4
  @dataclass
5
  class Config:
6
  # Model settings
7
- DEFAULT_MODEL: str = "HuggingFaceTB/SmolLM-135M-Instruct"
8
  DEVICE: str = "cpu" # Force CPU usage
9
 
10
  # Generation settings
 
4
  @dataclass
5
  class Config:
6
  # Model settings
7
+ DEFAULT_MODEL: str = "HuggingFaceTB/SmolLM2-135M-Instruct"
8
  DEVICE: str = "cpu" # Force CPU usage
9
 
10
  # Generation settings