Text Generation
Transformers
Safetensors
English
smartcoder_moe
Mixture of Experts
starcoder2
mixture-of-experts
code
smartcoder
conversational
custom_code
Instructions to use Johnblick187/SmartCoderMoE with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Johnblick187/SmartCoderMoE with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="Johnblick187/SmartCoderMoE", trust_remote_code=True) messages = [ {"role": "user", "content": "Who are you?"}, ] pipe(messages)# Load model directly from transformers import AutoModelForCausalLM model = AutoModelForCausalLM.from_pretrained("Johnblick187/SmartCoderMoE", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use Johnblick187/SmartCoderMoE with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "Johnblick187/SmartCoderMoE" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Johnblick187/SmartCoderMoE", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker
docker model run hf.co/Johnblick187/SmartCoderMoE
- SGLang
How to use Johnblick187/SmartCoderMoE with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "Johnblick187/SmartCoderMoE" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Johnblick187/SmartCoderMoE", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "Johnblick187/SmartCoderMoE" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Johnblick187/SmartCoderMoE", "messages": [ { "role": "user", "content": "What is the capital of France?" } ] }' - Docker Model Runner
How to use Johnblick187/SmartCoderMoE with Docker Model Runner:
docker model run hf.co/Johnblick187/SmartCoderMoE
Update modeling_smartcoder_moe.py
Browse files- modeling_smartcoder_moe.py +1 -10
modeling_smartcoder_moe.py
CHANGED
|
@@ -341,16 +341,7 @@ def load_smartcoder_moe(model_id="Johnblick187/SmartCoderMoE", dtype=torch.bfloa
|
|
| 341 |
sf_files = sorted(Path(model_dir).glob("*.safetensors"))
|
| 342 |
state_dict = {}
|
| 343 |
for f in sf_files:
|
| 344 |
-
state_dict.update(load_file(str(f)))
|
| 345 |
-
|
| 346 |
-
# Expert key remap (.weight suffix in the checkpoint vs raw Parameter
|
| 347 |
-
# here) is now handled by SmartCoderMoEMLP._load_from_state_dict
|
| 348 |
-
# itself, so load_state_dict() needs no manual remapping here anymore.
|
| 349 |
-
missing, unexpected = model.load_state_dict(state_dict, strict=False)
|
| 350 |
-
if missing:
|
| 351 |
-
print(f"Missing: {missing[:3]}{'...' if len(missing)>3 else ''}")
|
| 352 |
-
if unexpected:
|
| 353 |
-
print(f"Unexpected: {unexpected[:3]}{'...' if len(unexpected)>3 else ''}")
|
| 354 |
|
| 355 |
model = model.to(dtype)
|
| 356 |
print(f"Loaded! Params: {sum(p.numel() for p in model.parameters())/1e9:.2f}B")
|
|
|
|
| 341 |
sf_files = sorted(Path(model_dir).glob("*.safetensors"))
|
| 342 |
state_dict = {}
|
| 343 |
for f in sf_files:
|
| 344 |
+
state_dict.update(load_file(str(f))))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 345 |
|
| 346 |
model = model.to(dtype)
|
| 347 |
print(f"Loaded! Params: {sum(p.numel() for p in model.parameters())/1e9:.2f}B")
|