How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="anonymous192837192873/GraphGen-14b")
messages = [
    {"role": "user", "content": "Who are you?"},
]
pipe(messages)
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("anonymous192837192873/GraphGen-14b")
model = AutoModelForCausalLM.from_pretrained("anonymous192837192873/GraphGen-14b", device_map="auto")
messages = [
    {"role": "user", "content": "Who are you?"},
]
inputs = tokenizer.apply_chat_template(
	messages,
	add_generation_prompt=True,
	tokenize=True,
	return_dict=True,
	return_tensors="pt",
).to(model.device)

outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[-1]:]))
Quick Links

GraphGen-14b

Anonymous submission accompanying a conference paper.

Fine-tuned from Qwen/Qwen2.5-Coder-14B-Instruct on the companion GraphGen dataset to generate multi-agent workflow configurations from natural-language queries.

Quick start

from transformers import AutoModelForCausalLM, AutoTokenizer

repo = "anonymus192837192873/GraphGen-14b"
tok = AutoTokenizer.from_pretrained(repo)
model = AutoModelForCausalLM.from_pretrained(repo, torch_dtype="bfloat16", device_map="auto")

Notes

Released for anonymous peer review. Author and affiliation information is intentionally withheld.

Downloads last month
15
Safetensors
Model size
15B params
Tensor type
BF16
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for anonymous192837192873/GraphGen-14b

Base model

Qwen/Qwen2.5-14B
Finetuned
(127)
this model
Quantizations
1 model

Collection including anonymous192837192873/GraphGen-14b