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)):