Hugging Face's logo

manbeast3b
/
perfbench7test

Model card Files
xet
 Community
manbeast3b commited on
Commit
1781e0b
·
verified ·
1 Parent(s): c2a96d8

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -9,7 +9,7 @@ from torch import Generator
9
 
10
  Pipeline = None
11
  # Consistent environment variable setting
12
- os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128" # More robust memory management
13
 
14
  ckpt_id = "black-forest-labs/FLUX.1-schnell"
15
 
 
9
 
10
  Pipeline = None
11
  # Consistent environment variable setting
12
+ os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:1024" # More robust memory management
13
 
14
  ckpt_id = "black-forest-labs/FLUX.1-schnell"
15