Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-12-27 22:59:32 +01:00
Code Issues Packages Projects Releases Wiki Activity

Check for special characters in path on Windows (#81)

Browse Source 
Display warning message if detected
This commit is contained in:
jllllll 2023-06-17 17:09:22 -05:00 committed by GitHub
parent c42f183d3f
commit b2483e28d1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
webui.py
View File

@ -68,6 +68,15 @@ def check_env():
def install_dependencies():
# Check for special characters in installation path on Windows
if sys.platform.startswith("win"):
# punctuation contains: !"#$%&'()*+,-./:;<=>?@[\]^_`{|}~
from string import punctuation
# Allow some characters: _-:\/.'"
special_characters = punctuation.translate({ord(char): None for char in '_-:\\/.\'"'})
if any(char in script_dir for char in special_characters):
print_big_message("WARNING: Special characters were detected in the installation path!\n This can cause the installation to fail!")
# Select your GPU or, choose to run in CPU mode
print("What is your GPU")
print()