refactor: remove model zoo URLs

pykeio · Dec 28, 2023 · 153d7af · 153d7af
1 parent af97600
commit 153d7af
Show file tree

Hide file tree

Showing 15 changed files with 44 additions and 685 deletions.
diff --git a/examples/gpt2/examples/gpt2-no-ndarray.rs b/examples/gpt2/examples/gpt2-no-ndarray.rs
@@ -4,7 +4,7 @@ use std::{
 	sync::Arc
 };
 
-use ort::{download::language::machine_comprehension::GPT2, inputs, CUDAExecutionProvider, GraphOptimizationLevel, Session};
+use ort::{inputs, CUDAExecutionProvider, GraphOptimizationLevel, Session};
 use rand::Rng;
 use tokenizers::Tokenizer;
 
@@ -36,7 +36,7 @@ fn main() -> ort::Result<()> {
 	let session = Session::builder()?
 		.with_optimization_level(GraphOptimizationLevel::Level1)?
 		.with_intra_threads(1)?
-		.with_model_downloaded(GPT2::GPT2LmHead)?;
+		.with_model_downloaded("https://parcel.pyke.io/v2/cdn/assetdelivery/ortrsv2/ex_models/gpt2.onnx")?;
 
 	// Load the tokenizer and encode the prompt into a sequence of tokens.
 	let tokenizer = Tokenizer::from_file(Path::new(env!("CARGO_MANIFEST_DIR")).join("data").join("tokenizer.json")).unwrap();

diff --git a/examples/gpt2/examples/gpt2.rs b/examples/gpt2/examples/gpt2.rs
@@ -4,7 +4,7 @@ use std::{
 };
 
 use ndarray::{array, concatenate, s, Array1, Axis};
-use ort::{download::language::machine_comprehension::GPT2, inputs, CUDAExecutionProvider, GraphOptimizationLevel, Session, Tensor};
+use ort::{inputs, CUDAExecutionProvider, GraphOptimizationLevel, Session, Tensor};
 use rand::Rng;
 use tokenizers::Tokenizer;
 
@@ -36,7 +36,7 @@ fn main() -> ort::Result<()> {
 	let session = Session::builder()?
 		.with_optimization_level(GraphOptimizationLevel::Level1)?
 		.with_intra_threads(1)?
-		.with_model_downloaded(GPT2::GPT2LmHead)?;
+		.with_model_downloaded("https://parcel.pyke.io/v2/cdn/assetdelivery/ortrsv2/ex_models/gpt2.onnx")?;
 
 	// Load the tokenizer and encode the prompt into a sequence of tokens.
 	let tokenizer = Tokenizer::from_file(Path::new(env!("CARGO_MANIFEST_DIR")).join("data").join("tokenizer.json")).unwrap();

diff --git a/src/download/language.rs b/src/download/language.rs
diff --git a/src/download/language/machine_comprehension.rs b/src/download/language/machine_comprehension.rs
diff --git a/src/download/mod.rs b/src/download/mod.rs
diff --git a/src/download/vision.rs b/src/download/vision.rs
diff --git a/src/download/vision/body_face_gesture_analysis.rs b/src/download/vision/body_face_gesture_analysis.rs
diff --git a/src/download/vision/domain_based_image_classification.rs b/src/download/vision/domain_based_image_classification.rs