--- language: - pt license: apache-2.0 library_name: transformers tags: - portuguese - brasil - gemma - portugues - instrucao datasets: - rhaymison/superset base_model: google/gemma-2b-it pipeline_tag: text-generation --- # gemma-portuguese-2b-luana
## Model description updated: 2024-04-10 20:06 The gemma-portuguese-2b model is a portuguese model trained with the superset dataset with 250,000 instructions. The model is mainly focused on text generation and instruction. The model was not trained on math and code tasks. The model is generalist with focus on understand portuguese inferences. With this fine tuning for portuguese, you can adjust the model for a specific field. ## How to Use ```python from transformers import AutoTokenizer, pipeline import torch model = "rhaymison/gemma-portuguese-luana-2b" tokenizer = AutoTokenizer.from_pretrained(model) pipeline = pipeline( "text-generation", model=model, model_kwargs={"torch_dtype": torch.bfloat16}, device="cuda", ) messages = [ { "role": "system", "content": "Abaixo está uma instrução que descreve uma tarefa, juntamente com uma entrada que fornece mais contexto. Escreva uma resposta que complete adequadamente o pedido." }, {"role": "user", "content": "Me conte sobre a ida do homem a Lua."}, ] prompt = pipeline.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True) outputs = pipeline( prompt, max_new_tokens=256, do_sample=True, temperature=0.2, top_k=50, top_p=0.95 ) print(outputs[0]["generated_text"][len(prompt):].replace("model","")) #A viagem à Lua foi um esforço monumental realizado pela Agência Espacial dos EUA entre 1969 e 1972. #Foi um marco significativo na exploração espacial e na ciência humana. #Aqui está uma visão geral de sua jornada: 1. O primeiro voo espacial humano foi o de Yuri Gagarin, que voou a Terra em 12 de abril de 1961. ``` ```python from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer2 = AutoTokenizer.from_pretrained("rhaymison/gemma-portuguese-tom-cat-2b-it") model2 = AutoModelForCausalLM.from_pretrained("rhaymison/gemma-portuguese-tom-cat-2b-it", device_map={"":0}) tokenizer2.pad_token = tokenizer2.eos_token tokenizer2.add_eos_token = True tokenizer2.add_bos_token, tokenizer2.add_eos_token tokenizer2.padding_side = "right" ``` ```python def format_template( question:str): system_prompt = "Abaixo está uma instrução que descreve uma tarefa, juntamente com uma entrada que fornece mais contexto. Escreva uma resposta que complete adequadamente o pedido." text = f"""