How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="CorticalStack/gemma-7b-ultrachat-sft")
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("CorticalStack/gemma-7b-ultrachat-sft")
model = AutoModelForCausalLM.from_pretrained("CorticalStack/gemma-7b-ultrachat-sft", device_map="auto")
Quick Links

gemma-7b-ultrachat-sft

gemma-7b-ultrachat-sft is an SFT fine-tuned version of google/gemma-7b using the stingning/ultrachat dataset.

Fine-tuning configuration

LoRA

  • LoRA r: 8
  • LoRA alpha: 16
  • LoRA dropout: 0.1

Training arguments

  • Epochs: 1
  • Batch size: 4
  • Gradient accumulation steps: 6
  • Optimizer: paged_adamw_32bit
  • Max steps: 100
  • Learning rate: 0.0002
  • Weight decay: 0.001
  • Learning rate scheduler type: constant
  • Max seq length: 2048
Downloads last month
54
Safetensors
Model size
9B params
Tensor type
F16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support