miniXelta_01 / README.md
Xelta's picture
Upload README.md
d29dd51 verified
metadata
language: es
datasets:
  - Xelta/response_mongo_text
metrics:
  - accuracy
tags:
  - llama
  - 4bit
  - lora
model-index:
  - name: llama-2-7b-miniXelta
    results:
      - task:
          type: text-generation
          name: Text Generation
        dataset:
          name: response_mongo_text
          type: Xelta/response_mongo_text
        metrics:
          - name: Accuracy
            type: accuracy
            value: 0.95

Llama-2-7b-miniXelta

Este es un modelo ajustado a partir del modelo Llama-2-7b-chat-hf utilizando LoRA y precisión de 4 bits. Ha sido entrenado con el conjunto de datos Xelta/response_mongo_text.

Uso

Puedes usar este modelo para generación de texto de la siguiente manera:

from transformers import AutoModelForCausalLM, AutoTokenizer

model_name = "username/llama-2-7b-miniXelta"

model = AutoModelForCausalLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

prompt = "Quiero inscribirme, soy Mattias y mi edad es 28 años"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))