Skip to content

Commit 4ff5ec2

Browse files
Update examples/flux-tensorrt/README.md
Co-authored-by: Sayak Paul <[email protected]>
1 parent a93116c commit 4ff5ec2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

examples/flux-tensorrt/README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -105,10 +105,10 @@ image.save("test_pipeline.png")
105105
Example: Transformer Only (Other Modules on Torch)
106106
```python
107107
pipeline = FluxPipelineTRT.from_pretrained(
108-
"black-forest-labs/FLUX.1-dev",
109-
torch_dtype=torch.bfloat16,
110-
engine_transformer=engine_transformer,
111-
)
108+
"black-forest-labs/FLUX.1-dev",
109+
torch_dtype=torch.bfloat16,
110+
engine_transformer=engine_transformer,
111+
)
112112
pipeline.to("cuda")
113113
```
114114

0 commit comments

Comments
 (0)