v0.1.5 tokio_stream

AUTOM77 · May 5, 2024 · 3d270ef · 3d270ef
1 parent e0f7795
commit 3d270ef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/hfd-cli/Cargo.toml b/hfd-cli/Cargo.toml
@@ -1,11 +1,12 @@
 [package]
 name = "hfd-cli"
-version = "0.1.4"
+version = "0.1.5"
 edition = "2021"
 
 [dependencies]
 tokio = { version = "1.37.0", features = ["full"] }
 clap = { version= "4.5.4", features=["derive"] }
+tokio-stream = "0.1.15"
 libhfd = { path = "../hfd" }
 
 [[bin]]

diff --git a/hfd-cli/src/cli.rs b/hfd-cli/src/cli.rs
@@ -1,11 +1,12 @@
 use tokio;
+use tokio_stream::StreamExt;
 
 #[tokio::main]
 async fn main() {
     let api = libhfd::api::tokio::Api::new().unwrap();
 
     let _filename = api
-        .model("ByteDance/Hyper-SD".to_string())
+        .dataset("ByteDance/Hyper-SD".to_string())
         .get("Hyper-SDXL-8steps-lora.safetensors")
         .await
         .unwrap();