Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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():
|