Ignore warnings on Colab

This commit is contained in:
oobabooga 2023-10-21 21:45:25 -07:00
parent 1a34927314
commit 2d1b3332e4

View File

@ -1,4 +1,5 @@
import argparse
import os
import sys
from collections import OrderedDict
from pathlib import Path
@ -189,6 +190,7 @@ for k in ['chat', 'notebook', 'no_stream']:
# Security warnings
if args.trust_remote_code:
logger.warning('trust_remote_code is enabled. This is dangerous.')
if not 'COLAB_GPU' in os.environ:
if args.share:
logger.warning("The gradio \"share link\" feature uses a proprietary executable to create a reverse tunnel. Use it with care.")
if any((args.listen, args.share)) and not any((args.gradio_auth, args.gradio_auth_path)):