Update README.md

This commit is contained in:
hjh0119
2025-07-07 03:27:08 +00:00
parent ddd07b32b8
commit 5b849bb3a9

View File

@ -70,12 +70,13 @@ messages = [
],
}
]
processor = AutoProcessor.from_pretrained(MODEL_PATH, use_fast=True)
model = Glm4vForConditionalGeneration.from_pretrained(
pretrained_model_name_or_path=MODEL_PATH,
torch_dtype=torch.bfloat16,
device_map="auto",
)
processor = AutoProcessor.from_pretrained(MODEL_PATH, use_fast=True)
inputs = processor.apply_chat_template(
messages,
tokenize=True,