tudormunteanu commited on
Commit
2ec8a0a
1 Parent(s): f39de59

fix model ids

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -10,13 +10,13 @@ def load_model(model_size: str = "32B"):
10
  Note: You'll need to replace these with actual HuggingFace model IDs
11
  """
12
  model_map = {
13
- "0.5B": "Qwen/Qwen-0.5B",
14
- "1.5B": "Qwen/Qwen-1.5B",
15
- "7B": "Qwen/Qwen-7B",
16
  # ... add other model sizes as needed
17
  }
18
 
19
- model_id = model_map.get(model_size, "Qwen/Qwen-7B") # default to 7B if size not found
20
 
21
  tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
22
  model = AutoModelForCausalLM.from_pretrained(
 
10
  Note: You'll need to replace these with actual HuggingFace model IDs
11
  """
12
  model_map = {
13
+ "0.5B": "Qwen/Qwen2.5-Coder-0.5B",
14
+ "1.5B": "Qwen/Qwen2.5-Coder-1.5B",
15
+ "7B": "Qwen/Qwen2.5-Coder-7B",
16
  # ... add other model sizes as needed
17
  }
18
 
19
+ model_id = model_map.get(model_size, "Qwen/Qwen2.5-Coder-7B")
20
 
21
  tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
22
  model = AutoModelForCausalLM.from_pretrained(