theprint's picture
Upload README.md with huggingface_hub
259702c verified
|
Raw
History Blame Contribute Delete
1.73 kB
metadata
base_model: unsloth/Qwen3.5-2B
datasets:
  - theprint/CreativeWriting-3k
tags:
  - fine-tuned
  - lora
  - sft
  - auto-sft
language:
  - en
library_name: transformers

CreativeWriter-v1-2B

A fine-tuned version of unsloth/Qwen3.5-2B trained on theprint CreativeWriting 3k data using Auto-SFT — an automated hyperparameter search and supervised fine-tuning pipeline.

The base model was adapted to follow the style and content of the theprint CreativeWriting 3k dataset. Expect improved performance on tasks similar to those represented in the training data.

Model Details

Property Value
Base model unsloth/Qwen3.5-2B
Training data theprint/CreativeWriting-3k
Fine-tuning epochs 2
Fine-tuning date 2026-08-06
Fine-tuning method LoRA (merged to full 16-bit)

Training Hyperparameters

LoRA

Parameter Value
r 4
alpha 4
dropout 0.04
target_modules ['q_proj', 'v_proj', 'k_proj', 'o_proj', 'gate_proj', 'up_proj', 'down_proj']

Training

Parameter Value
learning_rate 1e-05
batch_size 2
gradient_accumulation_steps 2
warmup_ratio 0.05
max_seq_length 2048
quantization none

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

model     = AutoModelForCausalLM.from_pretrained("theprint/CreativeWriter-v1-2B")
tokenizer = AutoTokenizer.from_pretrained("theprint/CreativeWriter-v1-2B")

Generated by Auto-SFT — automated LoRA fine-tuning with hyperparameter search.