frankenstein-2.0 / README.md
Questionmarkboy's picture
Update README.md
0edf16a verified
|
Raw
History Blame Contribute Delete
2.46 kB
metadata
license: apache-2.0
pipeline_tag: text-generation
library_name: transformers
tags:
  - mistral
  - merge
  - lora
  - fine-tuned
  - qlora
  - chat
  - reasoning
base_model:
  - HuggingFaceH4/zephyr-7b-beta
  - teknium/OpenHermes-2.5-Mistral-7B
  - cognitivecomputations/Dolphin-2.6-mistral-7b-dpo

🧟 Frankenstein 2.0

A custom AI assistant stitched together from three Mistral-7B models, then fine-tuned β€” built entirely on Kaggle.

🧬 What Is This?

Frankenstein 2.0 is a merged model combining the strengths of three open-source models, then fine-tuned with QLoRA on coding + general instruction data.

Component Contribution
Zephyr-7B-beta Instruction following + structure
OpenHermes-2.5 Warmth + conversational tone
Dolphin-2.6-dpo Obedience + helpfulness

πŸŽ“ Training Details

  • Base: Merge of 3Γ— Mistral-7B variants
  • Fine-tuning: QLoRA (4-bit NF4)
  • LoRA: rank 16, alpha 32
  • Data: 3000 general + 3000 coding examples
  • Hardware: 2Γ— NVIDIA Tesla T4 (Kaggle)

πŸ’‘ Capabilities

βœ… Python coding & debugging
βœ… Explaining complex topics simply
βœ… Step-by-step reasoning (with <think> tags)
βœ… Document Q&A (RAG-ready)
βœ… Tool use (web search, calculator)

πŸš€ How To Use

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model = AutoModelForCausalLM.from_pretrained(
    "Questionmarkboy/frankenstein-2.0",
    torch_dtype=torch.float16,
    device_map="auto"
)
tokenizer = AutoTokenizer.from_pretrained("Questionmarkboy/frankenstein-2.0")

messages = [{"role": "user", "content": "Explain blockchain to a 10-year-old"}]
inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True,
    return_tensors="pt").to(model.device)
output = model.generate(inputs, max_new_tokens=200)
print(tokenizer.decode(output[0], skip_special_tokens=True))

⚠️ Limitations

  • 7B model β€” may struggle with very long code generation (e.g. full HTML apps)
  • Knowledge limited to training data
  • Best for English tasks

πŸ™ Credits

πŸ“œ License

Apache 2.0 (inherited from Mistral base). Please credit the original base models when using this.


It's alive! 🧟⚑