function_calling-gemma-2b

function_calling-gemma-2b is an SFT fine-tuned version of google/gemma-2b using a custom training dataset. This model was made with Phinetune

Process

  • Learning Rate: 1.41e-05
  • Maximum Sequence Length: 4096
  • Dataset: Inishds/function_calling
  • Split: train

πŸ’» Usage

!pip install -qU transformers
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline

model = "Inishds/function_calling-gemma-2b"
tokenizer = AutoTokenizer.from_pretrained(model)

# Example prompt
prompt = "Your example prompt here"

# Generate a response
model = AutoModelForCausalLM.from_pretrained(model)
pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
outputs = pipeline(prompt, max_length=50, num_return_sequences=1)
print(outputs[0]["generated_text"])
Downloads last month
5
Safetensors
Model size
3.03B params
Tensor type
FP16
Β·
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.