mistral-nouns500 / README.md
waterdrops0's picture
Update README.md
a08c9c5 verified
metadata
license: cc0-1.0
library_name: transformers
pipeline_tag: text-generation
tags:
  - text-generation
  - causal-lm
  - mistral
  - fine-tuned
language: en
base_model: mistralai/Mistral-7B-v0.1

Mistral-7B Fine-Tuned on Nouns DAO Comments

This model is a fine-tuned version of mistralai/Mistral-7B-v0.1 on a custom dataset of Nouns DAO comments.

Model Description

The model has been fine-tuned to generate comments by Nouns DAO members on proposals, focusing on community discussions and insights.

Intended Use

  • Primary Use Case: Generating realistic DAO member comments for proposals.
  • Languages Supported: English (en).

How to Use

You can use this model with the Transformers library:

from transformers import AutoTokenizer, AutoModelForCausalLM

model_name = "your_username/your_model_name"

tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name).to("cuda" or "cpu")

prompt = "The following is a comment by a Nouns DAO member on the proposal to gift 1000 glasses to kids in need:"

inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=100, repetition_penalty=1.15, temperature=0.7, top_p=0.9)

generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(generated_text)