llama

atoma-network · Mar 28, 2024 · 430ffbb · 430ffbb
1 parent 4205697
commit 430ffbb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/atoma-inference/src/candle/llama.rs b/atoma-inference/src/candle/llama.rs
@@ -153,7 +153,7 @@ pub fn run(prompt: String, cfg: Config) -> Result<String> {
     for (i, tensor) in result.iter().enumerate() {
         save_tensor_to_file(
             tensor,
-            &PathBuf::from(format!("llama_logits_{}.pt", i.to_string())),
+            format!("llama_logits_{}.pt", i.to_string()).as_str(),
         )
         .unwrap();
     }

diff --git a/atoma-inference/src/main.rs b/atoma-inference/src/main.rs
@@ -27,8 +27,8 @@ fn llama1() {
 }
 
 fn main() {
-    stable_diffusion();
-    // llama1();
+    // stable_diffusion();
+    llama1();
     // let result = llama::run("One day I will").unwrap();
     // println!("{}", result);