Resolving Interference When Merging Models
Paper • 2306.01708 • Published • 19
How to use vonjack/gemma2-2b-merged with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-generation", model="vonjack/gemma2-2b-merged")
messages = [
{"role": "user", "content": "Who are you?"},
]
pipe(messages) # Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("vonjack/gemma2-2b-merged")
model = AutoModelForCausalLM.from_pretrained("vonjack/gemma2-2b-merged")
messages = [
{"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
messages,
add_generation_prompt=True,
tokenize=True,
return_dict=True,
return_tensors="pt",
).to(model.device)
outputs = model.generate(**inputs, max_new_tokens=40)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))How to use vonjack/gemma2-2b-merged with llama-cpp-python:
# !pip install llama-cpp-python from llama_cpp import Llama llm = Llama.from_pretrained( repo_id="vonjack/gemma2-2b-merged", filename="gemma-2-2B-merged-Q4_K_M.gguf", )
llm.create_chat_completion(
messages = [
{
"role": "user",
"content": "What is the capital of France?"
}
]
)How to use vonjack/gemma2-2b-merged with llama.cpp:
brew install llama.cpp # Start a local OpenAI-compatible server with a web UI: llama-server -hf vonjack/gemma2-2b-merged:Q4_K_M # Run inference directly in the terminal: llama-cli -hf vonjack/gemma2-2b-merged:Q4_K_M
winget install llama.cpp # Start a local OpenAI-compatible server with a web UI: llama-server -hf vonjack/gemma2-2b-merged:Q4_K_M # Run inference directly in the terminal: llama-cli -hf vonjack/gemma2-2b-merged:Q4_K_M
# Download pre-built binary from: # https://github.com/ggerganov/llama.cpp/releases # Start a local OpenAI-compatible server with a web UI: ./llama-server -hf vonjack/gemma2-2b-merged:Q4_K_M # Run inference directly in the terminal: ./llama-cli -hf vonjack/gemma2-2b-merged:Q4_K_M
git clone https://github.com/ggerganov/llama.cpp.git cd llama.cpp cmake -B build cmake --build build -j --target llama-server llama-cli # Start a local OpenAI-compatible server with a web UI: ./build/bin/llama-server -hf vonjack/gemma2-2b-merged:Q4_K_M # Run inference directly in the terminal: ./build/bin/llama-cli -hf vonjack/gemma2-2b-merged:Q4_K_M
docker model run hf.co/vonjack/gemma2-2b-merged:Q4_K_M
How to use vonjack/gemma2-2b-merged with vLLM:
# Install vLLM from pip:
pip install vllm
# Start the vLLM server:
vllm serve "vonjack/gemma2-2b-merged"
# Call the server using curl (OpenAI-compatible API):
curl -X POST "http://localhost:8000/v1/chat/completions" \
-H "Content-Type: application/json" \
--data '{
"model": "vonjack/gemma2-2b-merged",
"messages": [
{
"role": "user",
"content": "What is the capital of France?"
}
]
}'docker model run hf.co/vonjack/gemma2-2b-merged:Q4_K_M
How to use vonjack/gemma2-2b-merged with SGLang:
# Install SGLang from pip:
pip install sglang
# Start the SGLang server:
python3 -m sglang.launch_server \
--model-path "vonjack/gemma2-2b-merged" \
--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": "vonjack/gemma2-2b-merged",
"messages": [
{
"role": "user",
"content": "What is the capital of France?"
}
]
}'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 "vonjack/gemma2-2b-merged" \
--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": "vonjack/gemma2-2b-merged",
"messages": [
{
"role": "user",
"content": "What is the capital of France?"
}
]
}'How to use vonjack/gemma2-2b-merged with Ollama:
ollama run hf.co/vonjack/gemma2-2b-merged:Q4_K_M
How to use vonjack/gemma2-2b-merged with Unsloth Studio:
curl -fsSL https://unsloth.ai/install.sh | sh # Run unsloth studio unsloth studio -H 0.0.0.0 -p 8888 # Then open http://localhost:8888 in your browser # Search for vonjack/gemma2-2b-merged to start chatting
irm https://unsloth.ai/install.ps1 | iex # Run unsloth studio unsloth studio -H 0.0.0.0 -p 8888 # Then open http://localhost:8888 in your browser # Search for vonjack/gemma2-2b-merged to start chatting
# No setup required # Open https://huggingface.co/spaces/unsloth/studio in your browser # Search for vonjack/gemma2-2b-merged to start chatting
How to use vonjack/gemma2-2b-merged with Docker Model Runner:
docker model run hf.co/vonjack/gemma2-2b-merged:Q4_K_M
How to use vonjack/gemma2-2b-merged with Lemonade:
# Download Lemonade from https://lemonade-server.ai/ lemonade pull vonjack/gemma2-2b-merged:Q4_K_M
lemonade run user.gemma2-2b-merged-Q4_K_M
lemonade list
winget install llama.cpp
# Start a local OpenAI-compatible server with a web UI:
llama-server -hf vonjack/gemma2-2b-merged:Q4_K_M# Run inference directly in the terminal:
llama-cli -hf vonjack/gemma2-2b-merged:Q4_K_M# Download pre-built binary from:
# https://github.com/ggerganov/llama.cpp/releases# Start a local OpenAI-compatible server with a web UI:
./llama-server -hf vonjack/gemma2-2b-merged:Q4_K_M# Run inference directly in the terminal:
./llama-cli -hf vonjack/gemma2-2b-merged:Q4_K_Mgit clone https://github.com/ggerganov/llama.cpp.git
cd llama.cpp
cmake -B build
cmake --build build -j --target llama-server llama-cli# Start a local OpenAI-compatible server with a web UI:
./build/bin/llama-server -hf vonjack/gemma2-2b-merged:Q4_K_M# Run inference directly in the terminal:
./build/bin/llama-cli -hf vonjack/gemma2-2b-merged:Q4_K_Mdocker model run hf.co/vonjack/gemma2-2b-merged:Q4_K_MThis is a merge of pre-trained language models created using mergekit.
This model was merged using the TIES merge method using google/gemma-2-2b as a base.
The following models were included in the merge:
The following YAML configuration was used to produce this model:
models:
- model: google/gemma-2-2b-it
parameters:
weight: 1
merge_method: ties
base_model: google/gemma-2-2b
parameters:
normalize: true
int8_mask: true
dtype: bfloat16
Install from brew
# Start a local OpenAI-compatible server with a web UI: llama-server -hf vonjack/gemma2-2b-merged:Q4_K_M# Run inference directly in the terminal: llama-cli -hf vonjack/gemma2-2b-merged:Q4_K_M