Update README.md
Browse files
README.md
CHANGED
|
@@ -12,9 +12,7 @@ base_model_relation: merge
|
|
| 12 |
pipeline_tag: text-to-image
|
| 13 |
---
|
| 14 |
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
## **Memory-efficient merging code (Flux.1-Dev + Flux.1-Krea-Dev)**
|
| 18 |
|
| 19 |
```py
|
| 20 |
%%capture
|
|
@@ -25,10 +23,13 @@ pipeline_tag: text-to-image
|
|
| 25 |
!pip install safetensors huggingface_hub hf_xet
|
| 26 |
```
|
| 27 |
|
|
|
|
|
|
|
| 28 |
```
|
| 29 |
from huggingface_hub import notebook_login, HfApi
|
| 30 |
notebook_login()
|
| 31 |
```
|
|
|
|
| 32 |
|
| 33 |
```py
|
| 34 |
from diffusers import FluxTransformer2DModel
|
|
@@ -102,8 +103,6 @@ model.to(torch.bfloat16).save_pretrained("merged/transformer")
|
|
| 102 |
api = HfApi()
|
| 103 |
repo_id = "prithivMLmods/Flux.1-krea-Merge-Transformer"
|
| 104 |
|
| 105 |
-
|
| 106 |
-
|
| 107 |
api.upload_folder(
|
| 108 |
folder_path="merged/",
|
| 109 |
path_in_repo=".",
|
|
@@ -111,4 +110,12 @@ api.upload_folder(
|
|
| 111 |
repo_type="model",
|
| 112 |
revision="main"
|
| 113 |
)
|
| 114 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
pipeline_tag: text-to-image
|
| 13 |
---
|
| 14 |
|
| 15 |
+
## **Sub-Memory-efficient merging code (Flux.1-Dev + Flux.1-Krea-Dev)**
|
|
|
|
|
|
|
| 16 |
|
| 17 |
```py
|
| 18 |
%%capture
|
|
|
|
| 23 |
!pip install safetensors huggingface_hub hf_xet
|
| 24 |
```
|
| 25 |
|
| 26 |
+
---
|
| 27 |
+
|
| 28 |
```
|
| 29 |
from huggingface_hub import notebook_login, HfApi
|
| 30 |
notebook_login()
|
| 31 |
```
|
| 32 |
+
---
|
| 33 |
|
| 34 |
```py
|
| 35 |
from diffusers import FluxTransformer2DModel
|
|
|
|
| 103 |
api = HfApi()
|
| 104 |
repo_id = "prithivMLmods/Flux.1-krea-Merge-Transformer"
|
| 105 |
|
|
|
|
|
|
|
| 106 |
api.upload_folder(
|
| 107 |
folder_path="merged/",
|
| 108 |
path_in_repo=".",
|
|
|
|
| 110 |
repo_type="model",
|
| 111 |
revision="main"
|
| 112 |
)
|
| 113 |
+
```
|
| 114 |
+
|
| 115 |
+
## For more information, visit the documentation.
|
| 116 |
+
|
| 117 |
+
> Flux is a suite of state-of-the-art text-to-image generation models based on diffusion transformers, developed by Black Forest Labs. The models are designed for high-quality generative image tasks, including text-to-image, inpainting, outpainting, and advanced structure or depth-controlled workflows. Flux is available through the Hugging Face diffusers library.
|
| 118 |
+
|
| 119 |
+
For detailed guides, examples, and API refer to:
|
| 120 |
+
- **[Main Flux Pipeline Documentation](https://huggingface.co/docs/diffusers/main/en/api/pipelines/flux)**
|
| 121 |
+
- **[Flux Transformer Model Documentation](https://huggingface.co/docs/diffusers/main/en/api/models/flux_transformer)**
|