Skip to content

Commit 62a037d

Browse files
committed
Remove nop line
1 parent 7dc9b96 commit 62a037d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

sharktank/sharktank/examples/export_paged_llm_v1.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -116,10 +116,9 @@ def setup_cache(model, shard_count):
116116
page_count=hp.context_length // llama_config.block_seq_stride
117117
)
118118
page_dim = torch.export.Dim("page")
119-
dynamic_shapes = [{0: page_dim}]
120119

120+
dynamic_shapes = [{0: page_dim}]
121121
unpacked = cache_state
122-
dynamic_shapes = dynamic_shapes
123122
arg_affinities = {}
124123
shard_dim = None
125124

0 commit comments

Comments
 (0)