Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Lint

Browse Source
This commit is contained in:
oobabooga 2023-10-23 13:09:03 -07:00
parent 280ae720d7
commit 1edf321362
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
modules/llama_attn_hijack.py
View File

@ -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

2
modules/shared.py
View File

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