Downloads · 30 days
3
23% of all-time downloads
ncterry1/local-llm-lab
local-llm-lab is a machine learning model from ncterry1. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
This model is a custom fine-tuned LLaMA-7B model for knowledge graph summarization.
Downloads · 30 days
3
23% of all-time downloads
All-time downloads
13
Public
Repo size
—
Likes
0
Public
Click a slice to open those files.
Other1.5 KB · 47%
From the Hugging Face model README
This model is a custom fine-tuned LLaMA-7B model for knowledge graph summarization.
from transformers import AutoModelForCausalLM, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("ncterry1/local-llm-lab")
model = AutoModelForCausalLM.from_pretrained("ncterry1/local-llm-lab")
prompt = "Explain a knowledge graph in one sentence."
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(inputs.input_ids, max_new_tokens=50)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))