Update README.md
Browse files
README.md
CHANGED
|
@@ -34,13 +34,13 @@ This compressed model was made from [rockerBOO/flux.1-dev-SRPO](https://huggingf
|
|
| 34 |
from diffusers import FluxPipeline, FluxTransformer2DModel
|
| 35 |
from dfloat11 import DFloat11Model
|
| 36 |
with no_init_weights():
|
| 37 |
-
|
| 38 |
-
|
| 39 |
-
|
| 40 |
-
|
| 41 |
-
|
| 42 |
-
|
| 43 |
-
|
| 44 |
|
| 45 |
pipe = FluxPipeline.from_pretrained(
|
| 46 |
"black-forest-labs/FLUX.1-dev",
|
|
|
|
| 34 |
from diffusers import FluxPipeline, FluxTransformer2DModel
|
| 35 |
from dfloat11 import DFloat11Model
|
| 36 |
with no_init_weights():
|
| 37 |
+
transformer = FluxTransformer2DModel.from_config(
|
| 38 |
+
FluxTransformer2DModel.load_config(
|
| 39 |
+
"black-forest-labs/FLUX.1-dev",
|
| 40 |
+
subfolder="transformer"
|
| 41 |
+
),
|
| 42 |
+
torch_dtype=torch.bfloat16
|
| 43 |
+
).to(torch.bfloat16)
|
| 44 |
|
| 45 |
pipe = FluxPipeline.from_pretrained(
|
| 46 |
"black-forest-labs/FLUX.1-dev",
|