Skip to content

Commit 075b4ba

Browse files
Di Xu (SWE)facebook-github-bot
authored andcommitted
Run ET-eager on message recall
Differential Revision: D83990682
1 parent f174974 commit 075b4ba

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

examples/models/llama/rope.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,9 @@ def __init__(self, params: ModelArgs):
235235
super().__init__()
236236
self.params = params
237237

238+
if not hasattr(self.params, "device"):
239+
self.params.device = "cpu"
240+
238241
# Choose the appropriate RoPE implementation
239242
if self.params.use_hf_rope:
240243
self.precompute_freqs_cis = partial(

0 commit comments

Comments
 (0)