huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix FLUX.1 weights (#2457)

Browse Source 
* fix FLUX.1 weights

* added flux1-dev.safetensors
This commit is contained in:
Eugene Hauptmann
2024-08-29 17:10:28 +02:00
committed by GitHub
parent 86613c00e2
commit c02b7c3272
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-examples/examples/flux/main.rs
View File

@ -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(),