multimodalart HF Staff commited on
Commit
c6731f9
·
verified ·
1 Parent(s): 332e802

Disable dynamo (TORCHDYNAMO_DISABLE) to silence outlines_core WON'T CONVERT spam

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -2,6 +2,9 @@ import os
2
  import sys
3
 
4
  os.environ.setdefault("PYTORCH_CUDA_ALLOC_CONF", "expandable_segments:True")
 
 
 
5
 
6
  # Bundled diffusers source: PR branch `ideogram4-prompt-enhancement` (YiYi's refactor + native upsampling).
7
  _HERE = os.path.dirname(os.path.abspath(__file__))
 
2
  import sys
3
 
4
  os.environ.setdefault("PYTORCH_CUDA_ALLOC_CONF", "expandable_segments:True")
5
+ # outlines_core ships an @torch.compile bitmask kernel dynamo can't trace (torch.device const) -> noisy
6
+ # WON'T CONVERT spam on every local upsample. We never use torch.compile at runtime, so disable dynamo.
7
+ os.environ.setdefault("TORCHDYNAMO_DISABLE", "1")
8
 
9
  # Bundled diffusers source: PR branch `ideogram4-prompt-enhancement` (YiYi's refactor + native upsampling).
10
  _HERE = os.path.dirname(os.path.abspath(__file__))