Pin the revision used for phi-v2 + make it the default. (#1572)

* Pin the revision used for phi-v2 + make it the default. * Tweak the custom-ops build.
huggingface · Jan 12, 2024 · 6242276 · 6242276
1 parent 4191518
commit 6242276
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/candle-examples/build.rs b/candle-examples/build.rs
@@ -27,11 +27,5 @@ fn main() -> Result<()> {
             bindings.write(kdir.rust_target).unwrap()
         }
     }
-    #[cfg(not(feature = "cuda"))]
-    {
-        for kdir in KERNEL_DIRS.iter() {
-            let _file = std::fs::File::create(kdir.rust_target)?;
-        }
-    }
     Ok(())
 }
diff --git a/candle-examples/examples/phi/main.rs b/candle-examples/examples/phi/main.rs
@@ -169,7 +169,7 @@ struct Args {
     #[arg(long)]
     model_id: Option<String>,
 
-    #[arg(long, default_value = "1.5")]
+    #[arg(long, default_value = "2")]
     model: WhichModel,
 
     #[arg(long)]
@@ -247,9 +247,8 @@ fn main() -> Result<()> {
                 match args.model {
                     WhichModel::V1 => "refs/pr/2".to_string(),
                     WhichModel::V1_5 => "refs/pr/18".to_string(),
-                    WhichModel::V2 | WhichModel::PuffinPhiV2 | WhichModel::PhiHermes => {
-                        "main".to_string()
-                    }
+                    WhichModel::V2 => "834565c23f9b28b96ccbeabe614dd906b6db551a".to_string(),
+                    WhichModel::PuffinPhiV2 | WhichModel::PhiHermes => "main".to_string(),
                 }
             }
         }