Downloads · 30 days
20
2% of all-time downloads
vlsp-2023-vllm/hoa-7b
hoa-7b is a text generation model from vlsp-2023-vllm. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as bigscience-bloom-rail-1.0.
Hoa is an autoregressive Large Language Model (LLM), based on Bloom's model architecture. Hoa was trained on part of the Common Crawl dataset in Vietnamese and English.
Downloads · 30 days
20
2% of all-time downloads
All-time downloads
1.1K
Public
Repo size
25 GB
Likes
0
Public
Click a slice to open those files.
.bin12.5 GB · 100%
From the Hugging Face model README
Hoa is an autoregressive Large Language Model (LLM), based on Bloom's model architecture. Hoa was trained on part of the Common Crawl dataset in Vietnamese and English.
Details will be available soon.
To contact us, mail to: [email protected] (Lê Anh Cường) | [email protected] (Hiếu) | [email protected] (Nguyễn Việt Cường)
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("vlsp-2023-vllm/hoa-7b")
model = AutoModelForCausalLM.from_pretrained("vlsp-2023-vllm/hoa-7b", low_cpu_mem_usage=True)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model.to(device)
prompt = "Địa chỉ trường Đại học Tôn Đức Thắng nằm ở số"
input_ids = tokenizer(prompt, return_tensors="pt")['input_ids'].to(device)
gen_tokens = model.generate(input_ids, max_length=max_length, repetition_penalty=1.1)
print(tokenizer.batch_decode(gen_tokens)[0])