Skip to content

Commit

Permalink
Add files via upload
Browse files Browse the repository at this point in the history
  • Loading branch information
BBC-Esq committed Aug 1, 2024
1 parent a425dee commit a436f9e
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions setup_gpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@

def install_libraries_with_retry(max_retries=3, delay=3):
libraries = [
# "accelerate==0.33.0",
"certifi==2024.7.4",
"charset-normalizer==3.3.2",
"colorama==0.4.6",
Expand All @@ -29,7 +28,6 @@ def install_libraries_with_retry(max_retries=3, delay=3):
"nvidia-cuda-runtime-cu12==12.1.105",
"nvidia-ml-py==12.555.43",
"openai-whisper==20231117",
# "optimum==1.21.2",
"packaging==24.1",
"platformdirs==4.2.2",
"psutil==6.0.0",
Expand Down Expand Up @@ -123,4 +121,4 @@ def main():
print(f"\033[92m\nTotal installation time: {int(hours):02d}:{int(minutes):02d}:{seconds:05.2f}\033[0m")

if __name__ == "__main__":
main()
main()

0 comments on commit a436f9e

Please sign in to comment.