walid-iguider's picture
Update README.md
dd76069 verified
|
raw
history blame
2.61 kB
metadata
license: apache-2.0
language:
  - it
pipeline_tag: text-generation
tags:
  - text-generation-inference
  - transformers
  - mistral
  - trl
  - sft
datasets:
  - mchl-labs/stambecco_data_it
widget:
  - text: >-
      Di seguito è riportata un'istruzione che descrive un'attività, abbinata ad
      un input che fornisce ulteriore informazione. Scrivi una risposta che
      soddisfi adeguatamente la richiesta. 

      ### Istruzione:

      Suggerisci un'attività serale romantica


      ### Input:


      ### Risposta:
    example_title: Example 1

Model Card for Minerva-3B-Instruct-v1.0

Minerva-3B-Instruct-v1.0 is an instruction-tuned version of the Minerva-3B-base-v1.0 model, specifically fine-tuned for understanding and following instructions in Italian.

Model Details

Model Description

This is the model card of a 🤗 transformers model that has been pushed on the Hub. This model card has been automatically generated.

Uses

Sample Code

  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
  import torch
  torch.random.manual_seed(0)
  # Run text generation pipeline with our next model
  prompt = """Di seguito è riportata un'istruzione che descrive un'attività, abbinata ad un input che fornisce
  ulteriore informazione. Scrivi una risposta che soddisfi adeguatamente la richiesta.
  
  ### Istruzione:
  Suggerisci un'attività serale romantica

  ### Input:
  
  
  ### Risposta:"""
  
  model_id = "walid-iguider/Minerva-3B-Instruct-v1.0"
  tokenizer = AutoTokenizer.from_pretrained(model_id)
  model = AutoModelForCausalLM.from_pretrained(
      model_id, 
      device_map="cuda", 
      torch_dtype="auto", 
      trust_remote_code=True, 
  )
  
  generation_args = {
      "max_new_tokens": 500,
      "return_full_text": False,
      "temperature": 0.0,
      "do_sample": False,
  }
  
  pipe = pipeline(
      "text-generation",
      model=model,
      tokenizer=tokenizer,
  )
  
  output = pipe(prompt, **generation_args)
  print(output[0]['generated_text'])