Update src/pipeline.py
Browse files- src/pipeline.py +1 -1
src/pipeline.py
CHANGED
|
@@ -14,7 +14,7 @@ from transformers import T5EncoderModel, CLIPTextModel
|
|
| 14 |
|
| 15 |
Pipeline: TypeAlias = FluxPipeline
|
| 16 |
torch.backends.cudnn.benchmark = True
|
| 17 |
-
torch.
|
| 18 |
|
| 19 |
CHECKPOINT = "jokerbit/flux.1-schnell-Robert-int8wo"
|
| 20 |
REVISION = "5ef0012f11a863e5111ec56540302a023bc8587b"
|
|
|
|
| 14 |
|
| 15 |
Pipeline: TypeAlias = FluxPipeline
|
| 16 |
torch.backends.cudnn.benchmark = True
|
| 17 |
+
torch.set_flush_denormal(True)
|
| 18 |
|
| 19 |
CHECKPOINT = "jokerbit/flux.1-schnell-Robert-int8wo"
|
| 20 |
REVISION = "5ef0012f11a863e5111ec56540302a023bc8587b"
|