From 64fb11a6b58c7abdfc3ac049c0554d4e8c2c1bc3 Mon Sep 17 00:00:00 2001
From: Hans <hans.chen@bricks.tools>
Date: Fri, 4 Oct 2024 15:57:58 +0800
Subject: [PATCH] fix: correct Sampler include

---
 src/sampler.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/sampler.rs b/src/sampler.rs
index 2ea42a9..d0d0e43 100644
--- a/src/sampler.rs
+++ b/src/sampler.rs
@@ -6,8 +6,8 @@ use neon::prelude::*;
 use neon::types::Finalize;
 
 cpp! {{
-  #include <executorch/examples/models/llama2/sampler/sampler.h>
-  #include <executorch/examples/models/llama2/sampler/sampler.cpp>
+  #include <executorch/extension/llm/sampler/sampler.h>
+  #include <executorch/extension/llm/sampler/sampler.cpp>
 }}
 
 cpp_class!(pub unsafe struct Sampler as "torch::executor::Sampler");