deepakkarkala/sft_sitcom_chandlerbing_jsonl
Viewer • Updated • 14.9k • 23
How to use deepakkarkala/gemma3_1b_lora_sft_sitcom with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-it")
model = PeftModel.from_pretrained(base_model, "deepakkarkala/gemma3_1b_lora_sft_sitcom")axolotl version: 0.10.0.dev0
adapter: qlora
base_model: google/gemma-3-1b-it
bf16: auto
chat_template: gemma3
datasets:
- path: deepakkarkala/sft_sitcom_chandlerbing_jsonl
split: train_without_fewshots
type: alpaca
ddp_find_unused_parameters: true
eval_sample_packing: false
evals_per_epoch: null
flash_attention: true
gradient_accumulation_steps: 8
gradient_checkpointing: true
gradient_checkpointing_kwargs:
use_reentrant: false
hub_model_id: deepakkarkala/gemma3_1b_lora_sft_sitcom
learning_rate: 0.0002
load_in_4bit: true
load_in_8bit: false
logging_steps: 1
lora_alpha: 16
lora_dropout: 0.05
lora_r: 32
lora_target_linear: true
lr_scheduler: cosine
micro_batch_size: 2
model_type: AutoModelForCausalLM
num_epochs: 4
optimizer: adamw_bnb_8bit
output_dir: ./outputs/out
pad_to_sequence_len: true
resume_from_checkpoint: null
sample_packing: true
saves_per_epoch: 1
sequence_len: 2048
special_tokens: null
tf32: true
tokenizer_type: AutoTokenizer
val_set_size: 0.05
wandb_entity: deepakkarkala-personal
wandb_log_model: checkpoint
wandb_name: sft_gemma3_1b
wandb_project: finetuning_llama31_8b_sitcom
wandb_run_id: sft_gemma3_1b_2
wandb_watch: null
warmup_ratio: 0.1
weight_decay: 0.0
This model is a fine-tuned version of google/gemma-3-1b-it on the deepakkarkala/sft_sitcom_chandlerbing_jsonl dataset.
More information needed
More information needed
More information needed
The following hyperparameters were used during training: