Skip to content

Commit 0ac82d0

Browse files
committed
Fix export
1 parent aa05b76 commit 0ac82d0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/models/llava/export_llava.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def quant_embedding(model):
186186
packed=False,
187187
).quantized_model()
188188

189-
quantized_token_embed = quant_embedding(llava.model_.language_model.model)
189+
quantized_token_embed = quant_embedding(llava.model_.model.language_model)
190190
token_dim_1 = Dim("token_dim_1", min=2, max=llava.text_model_args.max_seq_len)
191191
dynamic_shapes = [{1: token_dim_1}]
192192
with torch.no_grad():

0 commit comments

Comments
 (0)