Skip to content

Commit b3d6c8f

Browse files
committed
Add kwarg example inputs to eager model base
1 parent 2302e09 commit b3d6c8f

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)