fix: correct Sampler include

mybigday · Oct 4, 2024 · 64fb11a · 64fb11a
1 parent 96c2a37
commit 64fb11a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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");