Skip to content

Commit e530130

Browse files
committed
fix wrong import
1 parent dbabc9b commit e530130

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

examples/qualcomm/oss_scripts/llama/model/static_llama.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,8 @@
1212
import torch
1313
import torch.nn as nn
1414
import torch.nn.functional as F
15-
from executorch.examples.models.llama.llama_transformer import (
16-
ModelArgs,
17-
precompute_freqs_cis,
18-
)
15+
from executorch.examples.models.llama.model_args import ModelArgs
16+
from executorch.examples.models.llama.rope import precompute_freqs_cis
1917

2018

2119
def apply_rotary_emb_single(

0 commit comments

Comments
 (0)