jokerbit commited on
Commit
afab8a4
·
verified ·
1 Parent(s): 2978c8c

Upload src/pipeline.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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.comple, mode="max-autotune", fullgraph=True)
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