Downloads · 30 days
0
graphjudger
HaoyuHuang2/graphjudger
graphjudger is a graph machine learning model from HaoyuHuang2. Use it for the graph machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as mit.
This is the LoRA weights for the paper Can LLMs be Good Graph Judge for Knowledge Graph Construction?.
Access
Public
Updated Oct 2, 2025
Repo size
84.5 MB
Likes
3
Public
Repo makeup
Click a slice to open those files.
.safetensors50.4 MB · 97%
At a glance
- Task
- Graph Machine Learning
- Library
- transformers
- License
- mit
- Access
- Public
- Created
- Dec 6, 2024
- Updated
- Oct 2, 2025
- SHA
- 578facbf
Model card
From the Hugging Face model README
GraphJudger
This is the LoRA weights for the paper Can LLMs be Good Graph Judge for Knowledge Graph Construction?.
Code: https://github.com/hhy-huang/GraphJudge
Example
Loading:
BASE_MODEL = "models/llama-2-7b-hf"
LORA_WEIGHTS = "models/llama2-7b-lora-genwiki-context/"
model = LlamaForCausalLM.from_pretrained(
BASE_MODEL,
load_in_8bit=False
).half().cuda()
pipeline = transformers.pipeline (
"text-generation",
model=model,
tokenizer=tokenizer,
torch_dtype=torch.float16,
device=device
)
pipeline.model = PeftModel.from_pretrained(
model,
LORA_WEIGHTS,
torch_dtype=torch.float16,
).half().cuda()
- Task
- Graph Machine Learning
- Library
- transformers
- License
- mit
- Created
- Dec 6, 2024
- Updated
- Oct 2, 2025