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
|