Views
No views yet
| Parameters | Layers | Heads | Sequence Length | GQA num_key_value_heads* |
|---|---|---|---|---|
| 7.04 billion | 32 | 32 | 8192 | Variable |
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_name = "dataequity/DE-LM-7B"
5device = "cuda" # for GPU usage or "cpu" for CPU usage
6
7tokenizer = AutoTokenizer.from_pretrained(model_name)
8model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", trust_remote_code=True).to(device)
9
10inputs = tokenizer.encode("List the top 10 financial APIs", return_tensors="pt").to(device)
11outputs = model.generate(inputs, max_new_tokens=100, do_sample=True, top_p=0.95)
12print(tokenizer.decode(outputs[0]))
13
14# The model can also be used via the text-generation pipeline interface
15from transformers import pipeline
16generator = pipeline("text-generation", "dataequity/DE-LM-7B", torch_dtype="auto", trust_remote_code=True, device=device)
17outputs = generator("List the top 10 financial APIs", max_new_tokens=100, do_sample=True, top_p=0.95)
18print(outputs[0]["generated_text"])1@misc{DeciFoundationModels,
2title = {DeciLM-7B},
3author = {DeciAI Research Team},
4year = {2023}
5url={https://huggingface.co/Deci/DeciLM-7B},
6}