Text Generation
Transformers
Safetensors
English
ivme
language-model
transformer
rope
swiglu
muon
from-scratch
tiny
small
decoder-only
custom_code
Instructions to use IvmeLabs/Ivme-Conversate-v2-Base with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use IvmeLabs/Ivme-Conversate-v2-Base with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="IvmeLabs/Ivme-Conversate-v2-Base", trust_remote_code=True)# Load model directly from transformers import AutoModelForCausalLM model = AutoModelForCausalLM.from_pretrained("IvmeLabs/Ivme-Conversate-v2-Base", trust_remote_code=True, device_map="auto") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use IvmeLabs/Ivme-Conversate-v2-Base with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "IvmeLabs/Ivme-Conversate-v2-Base" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "IvmeLabs/Ivme-Conversate-v2-Base", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/IvmeLabs/Ivme-Conversate-v2-Base
- SGLang
How to use IvmeLabs/Ivme-Conversate-v2-Base 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 "IvmeLabs/Ivme-Conversate-v2-Base" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "IvmeLabs/Ivme-Conversate-v2-Base", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'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 "IvmeLabs/Ivme-Conversate-v2-Base" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "IvmeLabs/Ivme-Conversate-v2-Base", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use IvmeLabs/Ivme-Conversate-v2-Base with Docker Model Runner:
docker model run hf.co/IvmeLabs/Ivme-Conversate-v2-Base
Delete feedforward.py
Browse files- feedforward.py +0 -27
feedforward.py
DELETED
|
@@ -1,27 +0,0 @@
|
|
| 1 |
-
import torch
|
| 2 |
-
import torch.nn as nn
|
| 3 |
-
import torch.nn.functional as F
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
class SwiGLU(nn.Module):
|
| 7 |
-
"""SwiGLU feed-forward block (Section 4.6), as used in Llama/PaLM.
|
| 8 |
-
|
| 9 |
-
Standard formulation: down_proj(silu(gate_proj(x)) * up_proj(x))
|
| 10 |
-
The inner dim is scaled down from the naive 4x so that SwiGLU's extra
|
| 11 |
-
gate_proj matrix doesn't blow the parameter budget relative to a plain MLP
|
| 12 |
-
of the same nominal "4x" size -- this matches how Llama-style models size it.
|
| 13 |
-
"""
|
| 14 |
-
|
| 15 |
-
def __init__(self, hidden_dim: int, mult: float = 4.0):
|
| 16 |
-
super().__init__()
|
| 17 |
-
# standard correction: 4 * hidden * (2/3) keeps param count comparable
|
| 18 |
-
# to a plain (non-gated) 4x MLP, rounded to a clean multiple of 8.
|
| 19 |
-
inner_dim = int(hidden_dim * mult * 2 / 3)
|
| 20 |
-
inner_dim = ((inner_dim + 7) // 8) * 8
|
| 21 |
-
|
| 22 |
-
self.gate_proj = nn.Linear(hidden_dim, inner_dim, bias=False)
|
| 23 |
-
self.up_proj = nn.Linear(hidden_dim, inner_dim, bias=False)
|
| 24 |
-
self.down_proj = nn.Linear(inner_dim, hidden_dim, bias=False)
|
| 25 |
-
|
| 26 |
-
def forward(self, x: torch.Tensor) -> torch.Tensor:
|
| 27 |
-
return self.down_proj(F.silu(self.gate_proj(x)) * self.up_proj(x))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|