Lint

Vechtomov · Oct 23, 2023 · 1edf321 · 1edf321
1 parent 280ae72
commit 1edf321
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/modules/llama_attn_hijack.py b/modules/llama_attn_hijack.py
@@ -5,7 +5,6 @@
 import torch
 import torch.nn as nn
 
-
 import modules.shared as shared
 from modules.logging_colors import logger
 

diff --git a/modules/shared.py b/modules/shared.py
@@ -191,7 +191,7 @@
 # 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 'COLAB_GPU' not 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)):