Fredtt3 commited on
Commit
2a25462
·
verified ·
1 Parent(s): e10070c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -17,7 +17,8 @@ pipeline_tag: image-to-image
17
  library_name: diffusers
18
  ---
19
 
20
- > **Note:** This is a repackaged version of the [black-forest-labs/FLUX.2-dev](https://huggingface.co/black-forest-labs/FLUX.2-dev) model. The `flux2-dev.safetensors` file, which was part of the same model found in the `transformer/` folder, was removed because loading the model caused two transformers and OMM errors, even though the model theoretically fit on the GPU. The file size decreased from 178GB with the duplicated load to around 110GB when performing an inference.
 
21
 
22
  ![Teaser](./teaser_generation.png)
23
  ![Teaser](./teaser_editing.png)
 
17
  library_name: diffusers
18
  ---
19
 
20
+ > **Note:** This is a repackaging of the [black-forest-labs/FLUX.2-dev](https://huggingface.co/black-forest-labs/FLUX.2-dev) model. Only the `flux2-dev.safetensors` file located in the root directory was removed, as it contained the same model as the one defined in the `transformer/` folder. Because of this, `diffusers` was loading the transformer twice, causing out-of-memory (OOM) errors. By removing this file, the duplicate loading was avoided and memory usage during inference was reduced from approximately **178 GB** to **110 GB**, enabling stable execution.
21
+
22
 
23
  ![Teaser](./teaser_generation.png)
24
  ![Teaser](./teaser_editing.png)