Skip to content

Commit 20bccae

Browse files
committed
fix(chat_cli): update imports for refactored module structure
1 parent 28724cc commit 20bccae

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

examples/chat_cli.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,8 @@ def main():
286286
load_safetensors,
287287
)
288288
from pygpukit.llm.buffers import DecodeBuffers
289-
from pygpukit.llm.model import precompute_freqs_cis, sample_token
289+
from pygpukit.llm.layers import precompute_freqs_cis
290+
from pygpukit.llm.sampling import sample_token
290291
from pygpukit.ops.basic import kv_cache_prefill_gqa
291292

292293
# =========================================================================

0 commit comments

Comments
 (0)