Prefab supports v7.

Ai00-X · Jan 25, 2025 · 92181b9 · 92181b9
1 parent eac9022
commit 92181b9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/crates/ai00-core/src/lib.rs b/crates/ai00-core/src/lib.rs
@@ -425,7 +425,7 @@ async fn load_runtime(
                                 Runtime::new(context, bundle, reload, states, tokenizer).await
                             }
                         )+
-                        (version, _) => bail!("unsupported version: {:?}", version)
+                        // (version, _) => bail!("unsupported version: {:?}", version)
                     }
                 }
             }
@@ -435,9 +435,11 @@ async fn load_runtime(
                     (ModelVersion::V4, Precision::Fp16, v4::Model, v4::Bundle::<f16>),
                     (ModelVersion::V5, Precision::Fp16, v5::Model, v5::Bundle::<f16>),
                     (ModelVersion::V6, Precision::Fp16, v6::Model, v6::Bundle::<f16>),
+                    (ModelVersion::V7, Precision::Fp16, v7::Model, v7::Bundle::<f16>),
                     (ModelVersion::V4, Precision::Fp32, v4::Model, v4::Bundle::<f32>),
                     (ModelVersion::V5, Precision::Fp32, v5::Model, v5::Bundle::<f32>),
-                    (ModelVersion::V6, Precision::Fp32, v6::Model, v6::Bundle::<f32>)
+                    (ModelVersion::V6, Precision::Fp32, v6::Model, v6::Bundle::<f32>),
+                    (ModelVersion::V7, Precision::Fp32, v7::Model, v7::Bundle::<f32>)
                 }
             )
         }