Text Generation
Transformers
PyTorch
mistral
Not-For-All-Audiences
nsfw
Inference Endpoints
text-generation-inference
Edit model card

What is PetrolLM?

PetrolLM is Mistral-7B-v0.1 model fine-tune using QLoRA (4-bit precision) for the purposes of creative writing and roleplay.

The dataset consists of 5800 samples, with the composition as follows:

  • AICG Logs (~17%)
  • PygmalionAI/PIPPA (~17%)
  • Squish42/bluemoon-fandom-1-1-rp-cleaned (~13%)
  • OpenLeecher/Teatime (~2%)
  • Norquinal/claude_multiround_chat_1k (~17%)
  • jundurbin/airoboros-gpt4-1.4 (~17%)
  • totally-not-an-llm/EverythingLM-data-V2-sharegpt (~17%)

These samples were then back-filled using gpt-4/gpt-3.5-turbo-16k or otherwise converted to fit the prompt format.

Prompt Format

The model was finetuned with a prompt format similar to the original SuperHOT prototype: ```

style: roleplay characters: [char]: [description] summary: [scenario]

Format: [char]: [message] Human: [message] ```

Use in Text Generation Web UI

Install the bleeding-edge version of transformers from source:

pip install git+https://github.com/huggingface/transformers

Or, alternatively, change model_type in config.json from mistral to llama.

Use in SillyTavern UI

As an addendum, you can include one of the following as the Last Output Sequence:

Human: In your next reply, write at least two paragraphs. Be descriptive and immersive, providing vivid details about {{char}}'s actions, emotions, and the environment.
{{char}}:
{{char}} (2 paragraphs, engaging, natural, authentic, descriptive, creative):
[System note: Write at least two paragraphs. Be descriptive and immersive, providing vivid details about {{char}}'s actions, emotions, and the environment.]
{{char}}:

The third one seems to work the best. I would recommend experimenting with creating your own to best suit your needs.

Finetuing Parameters

  • LoRA Rank: 64
  • LoRA Alpha: 16
  • LoRA Dropout: 0.1
  • BF16 Training
  • Cutoff Length: 2048
  • Training Epoch(s): 2
Downloads last month
3

Datasets used to train Norquinal/PetrolLM

Collection including Norquinal/PetrolLM