Skip to content

Commit f502e65

Browse files
update
1 parent 624daaf commit f502e65

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

glm-edge-v/glmv_helper.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -319,7 +319,7 @@ def convert_glmv_model(model_id, output_dir, quantization_config):
319319
pkv = model.model(
320320
input_ids=input_ids,
321321
attention_mask=torch.ones((2, 2), dtype=torch.int64),
322-
)[1]
322+
images = torch.zeros([1, 3, image_size, image_size]))[1]
323323
model.forward = types.MethodType(_chatglm_transformer_forward, model)
324324

325325
model.config.torchscript = True

0 commit comments

Comments
 (0)