diff --git a/README.md b/README.md index 17f06b4..b70c5b3 100644 --- a/README.md +++ b/README.md @@ -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,