Hugging Face's logo

slobers
/
sos05

Model card Files
xet
Community
slobers commited on
Commit
a942e72
·
verified ·
1 Parent(s): af25da5

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -13,7 +13,7 @@ from diffusers import FluxTransformer2DModel, DiffusionPipeline
13
  from torchao.quantization import quantize_, int8_weight_only, fpx_weight_only
14
 
15
  os.environ['PYTORCH_CUDA_ALLOC_CONF']="expandable_segments:True"
16
- os.environ["TOKENIZERS_PARALLELISM"] = "False"
17
  torch._dynamo.config.suppress_errors = True
18
 
19
  Pipeline = None
 
13
  from torchao.quantization import quantize_, int8_weight_only, fpx_weight_only
14
 
15
  os.environ['PYTORCH_CUDA_ALLOC_CONF']="expandable_segments:True"
16
+ os.environ["TOKENIZERS_PARALLELISM"] = "True"
17
  torch._dynamo.config.suppress_errors = True
18
 
19
  Pipeline = None