Add a proper documentation (#3885)

This commit is contained in:
oobabooga 2023-10-21 19:15:54 -03:00 committed by GitHub
parent 5a5bc135e9
commit 6efb990b60
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
30 changed files with 707 additions and 932 deletions

View file

@ -182,6 +182,10 @@ def install_webui():
while use_cuda118 not in 'YN':
print("Invalid choice. Please try again.")
use_cuda118 = input("Input> ").upper().strip('"\'').strip()
if use_cuda118 == 'Y':
print(f"CUDA: 11.8")
else:
print(f"CUDA: 12.1")
install_pytorch = f"python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/{'cu121' if use_cuda118 == 'N' else 'cu118'}"
elif not is_macos() and choice == "B":