metadata
license: other
license_name: mocha
license_link: LICENSE
datasets:
- theguywhosucks/mocha
language:
- en
pipeline_tag: text-generation
tags:
- english
- composition
- sentance_completion
Mocha β
Mocha is a sentence completion model designed for lightweight, fast, and creative text generation. Built with efficiency in mind, Mocha allows you to integrate natural language completion into your projects without the overhead of larger models.
π₯ Features
- β‘ Lightweight β optimized for speed and deployment.
- π Sentence Completion β generate contextually relevant endings for text prompts.
- π¦ Safetensors Format β stored in
.safetensorsfor secure and efficient loading.
π Project Structure
.
βββ mocha.safetensors # The model weights
βββ config.json
βββ generation_config.json
βββ itos.json
βββ stoi.json
βββ logo.png # Project logo
βββ banner.png # Project banner
βββ README.md
π Usage
You can load Mocha with π€ Transformers and safetensors:
from transformers import AutoTokenizer, AutoModelForCausalLM
import torch
# Load tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("theguywhosucks/Mocha")
model = AutoModelForCausalLM.from_pretrained(
"theguywhosucks/Mocha",
torch_dtype=torch.float16
)
# Example usage
prompt = "The future of AI is"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=20)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
πΈ Assets
π οΈ Requirements
- Python 3.9+
- Transformers
- Safetensors
- PyTorch
Install dependencies:
pip install torch transformers safetensors
π License
This project is licensed under the Mocha Proprietary License. Usage, distribution, and modification are restricted. Please see the LICENSE file for full details.
β Mocha β lightweight sentence completion made simple.