How to use shuolucs/UniLayout-Reward with Transformers:
# Load model directly from transformers import LlavaLlamaForClass model = LlavaLlamaForClass.from_pretrained("shuolucs/UniLayout-Reward", dtype="auto")
Reward model checkpoint for UniLayout layout generation research.
Base model: liuhaotian/llava-v1.6-vicuna-7b. Vision tower: openai/clip-vit-large-patch14-336.
liuhaotian/llava-v1.6-vicuna-7b
openai/clip-vit-large-patch14-336
This repository contains the cleaned model config, tokenizer files, and sharded safetensors weights.
# Load model directly from transformers import LlavaLlamaForClass model = LlavaLlamaForClass.from_pretrained("shuolucs/UniLayout-Reward", dtype="auto")