Hugging Face's logo

slobers
/
Flux001

Model card Files
xet
 Community
slobers commited on
Commit
3e10352
·
verified ·
1 Parent(s): 6efbdc0

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -17,7 +17,7 @@ from para_attn.context_parallel.diffusers_adapters import parallelize_pipe
17
  os.environ['PYTORCH_CUDA_ALLOC_CONF']="expandable_segments:True"
18
  os.environ["TOKENIZERS_PARALLELISM"] = "True"
19
  torch._dynamo.config.suppress_errors = True
20
- torch.cuda.set_device(dist.get_rank())
21
  Pipeline = None
22
  ids = "slobers/Flux.1.Schnella"
23
  Revision = "e34d670e44cecbbc90e4962e7aada2ac5ce8b55b"
 
17
  os.environ['PYTORCH_CUDA_ALLOC_CONF']="expandable_segments:True"
18
  os.environ["TOKENIZERS_PARALLELISM"] = "True"
19
  torch._dynamo.config.suppress_errors = True
20
+ #torch.cuda.set_device(dist.get_rank())
21
  Pipeline = None
22
  ids = "slobers/Flux.1.Schnella"
23
  Revision = "e34d670e44cecbbc90e4962e7aada2ac5ce8b55b"