Skip to content

Commit 10cb7e7

Browse files
committed
Add kwarg example inputs to eager model base
1 parent 9be5f57 commit 10cb7e7

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/models/llama2/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -341,3 +341,4 @@ def _transform_for_pre_quantization(self, checkpoint):
341341
int(embedding_bit_width),
342342
embedding_group_size,
343343
)
344+

0 commit comments

Comments
 (0)