How to use shuolucs/UniLayout-Reward with Transformers:
# Load model directly from transformers import LlavaLlamaForClass model = LlavaLlamaForClass.from_pretrained("shuolucs/UniLayout-Reward", device_map="auto")
