NewEden/SDPrompter4b

🔍 Overview

This is the second in a line of models dedicated to creating Stable-Diffusion prompts when given a character appearance. Made for the CharGen Project, This has been finetuned ontop of Delta-Vector/Holland-4B-V1

⚖️ Quants

Available quantization formats:

• GGUF: https://huggingface.co/mradermacher/SDPrompter4b-GGUF
• EXL2: https://huggingface.co/

💬 Prompting

Recommended format: ChatML, Use the following system prompt for the model. I'd advise against setting a high amount of output tokens as the model loops, use 0.1 min-p and temp-1 to keep it coherent.

Create a prompt for Stable Diffusion based on the information below.

🌟 Credits

Finetuned on 1xRTX6000 provided by Kubernetes_bad, All credits goes to Kubernetes_bad, LucyKnada and the rest of Anthracite.

🛠️ Axolotl Config)

base_model: Delta-Vector/Holland-4B-V1
model_type: AutoModelForCausalLM
tokenizer_type: AutoTokenizer
trust_remote_code: true
load_in_8bit: false
load_in_4bit: false
strict: false
datasets:
  - path: NewEden/CivitAI-SD-Prompts
datasets:
  - path: NewEden/CivitAI-Prompts-Sharegpt
    type: chat_template
    chat_template: chatml
    roles_to_train: ["gpt"]
    field_messages: conversations
    message_field_role: from
    message_field_content: value
    train_on_eos: turn
dataset_prepared_path:
val_set_size: 0.02
output_dir: ./outputs/out2
sequence_len: 8192
sample_packing: true
eval_sample_packing: false
pad_to_sequence_len: true
plugins:
  - axolotl.integrations.liger.LigerPlugin
liger_rope: true
liger_rms_norm: true
liger_swiglu: true
liger_fused_linear_cross_entropy: true
wandb_project: SDprompter-final
wandb_entity:
wandb_watch:
wandb_name: SDprompter-final
wandb_log_model:
gradient_accumulation_steps: 16
micro_batch_size: 1
num_epochs: 4
optimizer: paged_adamw_8bit
lr_scheduler: cosine
learning_rate: 0.00001
train_on_inputs: false
group_by_length: false
bf16: auto
fp16:
tf32: true
gradient_checkpointing: true
gradient_checkpointing_kwargs:
  use_reentrant: false
early_stopping_patience:
resume_from_checkpoint:
local_rank:
logging_steps: 1
xformers_attention:
flash_attention: true
warmup_ratio: 0.05
evals_per_epoch: 4
saves_per_epoch: 1
debug:
weight_decay: 0.01
special_tokens:
  pad_token: <|finetune_right_pad_id|>
  eos_token: <|eot_id|>
auto_resume_from_checkpoints: true