From 1edf321362b3f5b3c2486fca8fd3931628c8b669 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Mon, 23 Oct 2023 13:09:03 -0700
Subject: [PATCH] Lint

---
 modules/llama_attn_hijack.py | 1 -
 modules/shared.py            | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/modules/llama_attn_hijack.py b/modules/llama_attn_hijack.py
index a7559a57..00436fb2 100644
--- a/modules/llama_attn_hijack.py
+++ b/modules/llama_attn_hijack.py
@@ -5,7 +5,6 @@ from typing import Optional, Tuple
 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
index 3744d551..626c2bf8 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -191,7 +191,7 @@ for k in ['chat', 'notebook', 'no_stream', 'mul_mat_q']:
 # 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)):