mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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
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
import torch.nn as nn import torch.nn as nn
import modules.shared as shared import modules.shared as shared
from modules.logging_colors import logger 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 # Security warnings
if args.trust_remote_code: if args.trust_remote_code:
logger.warning('trust_remote_code is enabled. This is dangerous.') 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: if args.share:
logger.warning("The gradio \"share link\" feature uses a proprietary executable to create a reverse tunnel. Use it with care.") 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)): if any((args.listen, args.share)) and not any((args.gradio_auth, args.gradio_auth_path)):