diff options
| author | Eugene Hauptmann <eugene.hp2012@gmail.com> | 2024-08-29 17:10:28 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-08-29 17:10:28 +0200 |
| commit | c02b7c327297f25fc9cc60b8f39c9aa32e40ff21 (patch) | |
| tree | 4a551c7c06b38b47c49e9050b537b264f022a0a7 /candle-examples/examples | |
| parent | 86613c00e216750f32a326dbff5cc993d5e0067e (diff) | |
| download | candle-c02b7c327297f25fc9cc60b8f39c9aa32e40ff21.tar.gz candle-c02b7c327297f25fc9cc60b8f39c9aa32e40ff21.tar.bz2 candle-c02b7c327297f25fc9cc60b8f39c9aa32e40ff21.zip | |
Fix FLUX.1 weights (#2457)
* fix FLUX.1 weights
* added flux1-dev.safetensors
Diffstat (limited to 'candle-examples/examples')
| -rw-r--r-- | candle-examples/examples/flux/main.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/candle-examples/examples/flux/main.rs b/candle-examples/examples/flux/main.rs index a9278d01..539ae6f2 100644 --- a/candle-examples/examples/flux/main.rs +++ b/candle-examples/examples/flux/main.rs @@ -147,8 +147,8 @@ fn run(args: Args) -> Result<()> { println!("CLIP\n{clip_emb}"); let img = { let model_file = match model { - Model::Schnell => bf_repo.get("flux1-schnell.sft")?, - Model::Dev => bf_repo.get("flux1-dev.sft")?, + Model::Schnell => bf_repo.get("flux1-schnell.safetensors")?, + Model::Dev => bf_repo.get("flux1-dev.safetensors")?, }; let vb = unsafe { VarBuilder::from_mmaped_safetensors(&[model_file], dtype, &device)? }; @@ -189,7 +189,7 @@ fn run(args: Args) -> Result<()> { println!("latent img\n{img}"); let img = { - let model_file = bf_repo.get("ae.sft")?; + let model_file = bf_repo.get("ae.safetensors")?; let vb = unsafe { VarBuilder::from_mmaped_safetensors(&[model_file], dtype, &device)? }; let cfg = match model { Model::Dev => flux::autoencoder::Config::dev(), |