Skip to content

Commit c501128

Browse files
committed
Update readme
1 parent c697806 commit c501128

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

+2-1
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ Examples with the Rust crate: [here](diffusion_rs_examples/examples).
7272
```rust
7373
use std::time::Instant;
7474

75-
use diffusion_rs_core::{DiffusionGenerationParams, ModelSource, Offloading, Pipeline, TokenSource};
75+
use diffusion_rs_core::{DiffusionGenerationParams, ModelSource, ModelDType, Offloading, Pipeline, TokenSource};
7676
use tracing::level_filters::LevelFilter;
7777
use tracing_subscriber::EnvFilter;
7878

@@ -87,6 +87,7 @@ let pipeline = Pipeline::load(
8787
TokenSource::CacheToken,
8888
None,
8989
None,
90+
&ModelDType::Auto,
9091
)?;
9192

9293
let start = Instant::now();

0 commit comments

Comments
 (0)