roman-bachmann commited on
Commit
d180dea
·
verified ·
1 Parent(s): 7d2453d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -16,6 +16,8 @@ from flextok.flextok_wrapper import FlexTokFromHub
16
  from flextok.utils.demo import imgs_from_urls, denormalize, batch_to_pil
17
  from flextok.utils.misc import detect_bf16_support, get_bf16_context, get_generator
18
 
 
 
19
 
20
  # We recommend running this demo on an A100 GPU
21
  if torch.cuda.is_available():
 
16
  from flextok.utils.demo import imgs_from_urls, denormalize, batch_to_pil
17
  from flextok.utils.misc import detect_bf16_support, get_bf16_context, get_generator
18
 
19
+ os.environ["TORCH_LOGS"] = "+dynamo"
20
+ os.environ["TORCHDYNAMO_VERBOSE"] = "1"
21
 
22
  # We recommend running this demo on an A100 GPU
23
  if torch.cuda.is_available():