Upload src/pipeline.py with huggingface_hub
Browse files- src/pipeline.py +1 -1
src/pipeline.py
CHANGED
|
@@ -12,7 +12,7 @@ from torchao.quantization import quantize_, int8_weight_only
|
|
| 12 |
from transformers import T5EncoderModel, CLIPTextModel, logging
|
| 13 |
from functools import partial
|
| 14 |
|
| 15 |
-
my_compile_max = partial(torch.
|
| 16 |
|
| 17 |
Pipeline: TypeAlias = FluxPipeline
|
| 18 |
torch.backends.cudnn.benchmark = True
|
|
|
|
| 12 |
from transformers import T5EncoderModel, CLIPTextModel, logging
|
| 13 |
from functools import partial
|
| 14 |
|
| 15 |
+
my_compile_max = partial(torch.compile, mode="max-autotune", fullgraph=True)
|
| 16 |
|
| 17 |
Pipeline: TypeAlias = FluxPipeline
|
| 18 |
torch.backends.cudnn.benchmark = True
|