FlorianVal commited on
Commit
2136da5
·
verified ·
1 Parent(s): 0d894b6

Fix Space Gradio dependency and default to Qwen demo

Browse files
Files changed (2) hide show
  1. app.py +6 -6
  2. requirements.txt +1 -1
app.py CHANGED
@@ -13,16 +13,16 @@ from src.inference import load_dssd_model, DSSDecoder, TokenInfo, StreamEvent, S
13
 
14
  # Available models configuration
15
  AVAILABLE_MODELS = {
16
- "DSSD-Llama3-8B": {
17
- "model_name": "meta-llama/Meta-Llama-3-8B",
18
- "repo_id": "valcore/DSSD-Llama3-8B",
19
- "local_path": "../checkpoints/llama3-8b-4bit",
20
- },
21
  "DSSD-Qwen3-0.6B": {
22
  "model_name": "Qwen/Qwen3-0.6B",
23
- "repo_id": "valcore/DSSD-Qwen3-0.6B",
24
  "local_path": "../checkpoints/qwen3-0.6b",
25
  },
 
 
 
 
 
26
  }
27
 
28
  # Color palette for exit heads (colorblind-friendly)
 
13
 
14
  # Available models configuration
15
  AVAILABLE_MODELS = {
 
 
 
 
 
16
  "DSSD-Qwen3-0.6B": {
17
  "model_name": "Qwen/Qwen3-0.6B",
18
+ "repo_id": "FlorianVal/DSSD-Qwen3-0.6B",
19
  "local_path": "../checkpoints/qwen3-0.6b",
20
  },
21
+ "DSSD-Llama3-8B": {
22
+ "model_name": "meta-llama/Meta-Llama-3-8B",
23
+ "repo_id": "FlorianVal/DSSD-Llama3-8B",
24
+ "local_path": "../checkpoints/llama3-8b-4bit",
25
+ },
26
  }
27
 
28
  # Color palette for exit heads (colorblind-friendly)
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
  torch>=2.0.0,<3.0.0
2
  transformers>=4.54.1,<4.55.0
3
- gradio>=4.0.0,<5.0.0
4
  bitsandbytes>=0.41.0
5
  accelerate>=0.25.0,<1.0.0
6
  huggingface_hub>=0.19.0
 
1
  torch>=2.0.0,<3.0.0
2
  transformers>=4.54.1,<4.55.0
3
+ gradio==6.3.0
4
  bitsandbytes>=0.41.0
5
  accelerate>=0.25.0,<1.0.0
6
  huggingface_hub>=0.19.0