This commit is contained in:
jamjamjon
2025-05-20 17:01:27 +08:00
parent 26de63d239
commit d3c738b5cf
110 changed files with 298 additions and 264 deletions

View File

@@ -1,5 +1,5 @@
use anyhow::Result;
use usls::{models::Clip, DataLoader, ModelConfig, Ops};
use usls::{models::Clip, Config, DataLoader, Ops};
#[derive(argh::FromArgs)]
/// CLIP Example
@@ -17,7 +17,7 @@ fn main() -> Result<()> {
let args: Args = argh::from_env();
// build model
let config = ModelConfig::jina_clip_v1()
let config = Config::jina_clip_v1()
.with_device_all(args.device.as_str().try_into()?)
.commit()?;
let mut model = Clip::new(config)?;