mingyi456 commited on
Commit
bb2e99b
·
verified ·
1 Parent(s): c50d7c1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -7
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
- 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",
 
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",