Downloads · 30 days
14
56% of all-time downloads
sethderrick/gemma-2-2B-it-thinking-function_calling-V0
gemma-2-2B-it-thinking-function_calling-V0 is a text generation model from sethderrick. Use it when you need the model to write or continue text. It is set up for peft. The card lists the license as apache-2.0.
This is a fine-tuned version of google/gemma-2-2b-it optimized for function calling with thinking.
Downloads · 30 days
14
56% of all-time downloads
All-time downloads
25
Public
Repo size
4.9 GB
Likes
0
Public
Click a slice to open those files.
.safetensors4.8 GB · 99%
From the Hugging Face model README
This is a fine-tuned version of google/gemma-2-2b-it optimized for function calling with thinking.
## Model Details
- Base model: google/gemma-2-2b-it
- Fine-tuned with LoRA for function calling capability
- Includes "thinking" step before function calls
## Usage
```python
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel, PeftConfig
model_name = "sethderrick/gemma-2-2B-it-thinking-function_calling-V0"
# Load the model
config = PeftConfig.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path)
model = PeftModel.from_pretrained(model, model_name)
# Load the tokenizer
tokenizer = AutoTokenizer.from_pretrained(model_name)
# Use for function calling
# ...
```