|
--- |
|
license: apache-2.0 |
|
datasets: |
|
- emre/llama-2-13b-code-chat |
|
tags: |
|
- code |
|
--- |
|
|
|
# π¦π» CodeLlama |
|
|
|
|
|
`emre/llama-2-13b-code-chat` is a Llama 2 version of [**CodeAlpaca**](https://github.com/sahil280114/codealpaca). |
|
|
|
## π§ Training |
|
|
|
This model is based on the `llama-2-13b-chat-hf` model, fine-tuned using QLoRA on the [`mlabonne/CodeLlama-2-20k`](https://huggingface.co/datasets/mlabonne/CodeLlama-2-20k) dataset. |
|
It was trained on an Colab Pro+It was trained Colab Pro+. |
|
It is mainly designed for educational purposes, not for inference but can be used exclusively with BBVA Group, GarantiBBVA and its subsidiaries. |
|
|
|
|
|
## π» Usage |
|
|
|
``` python |
|
# pip install transformers accelerate |
|
|
|
from transformers import AutoTokenizer |
|
import transformers |
|
import torch |
|
|
|
model = "emre/llama-2-13b-code-chat" |
|
prompt = "Write Python code to generate an array with all the numbers from 1 to 100" |
|
|
|
tokenizer = AutoTokenizer.from_pretrained(model) |
|
pipeline = transformers.pipeline( |
|
"text-generation", |
|
model=model, |
|
torch_dtype=torch.float16, |
|
device_map="auto", |
|
) |
|
|
|
sequences = pipeline( |
|
f'<s>[INST] {prompt} [/INST]', |
|
do_sample=True, |
|
top_k=10, |
|
num_return_sequences=1, |
|
eos_token_id=tokenizer.eos_token_id, |
|
max_length=200, |
|
) |
|
for seq in sequences: |
|
print(f"Result: {seq['generated_text']}") |
|
``` |
|
|
|
Ouput: |
|
``` |
|
Here is a Python code to generate an array with all the numbers from 1 to 100: |
|
|
|
β
``` |
|
numbers = [] |
|
for i in range(1,101): |
|
numbers.append(i) |
|
β
``` |
|
|
|
This code generates an array with all the numbers from 1 to 100 in Python. It uses a loop that iterates over the range of numbers from 1 to 100, and for each number, it appends that number to the array 'numbers'. The variable 'numbers' is initialized to a list, and its length is set to 101 by using the range of numbers (0-99). |
|
|
|
``` |