tsqn commited on
Commit
bc1d5c6
·
verified ·
1 Parent(s): f36dd01

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -1
utils.py CHANGED
@@ -224,4 +224,5 @@ class ProgressBar:
224
 
225
  def install_packages():
226
  subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "pip", "wheel", "setuptools"])
227
- subprocess.check_call([sys.executable, "-m", "pip", "install", "-pre", "torch", "torchvision", "torchaudio", "--index-url", "https://download.pytorch.org/whl/nightly/cu124"])
 
 
224
 
225
  def install_packages():
226
  subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "pip", "wheel", "setuptools"])
227
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "--pre", "torch", "torchvision", "torchaudio", "--index-url", "https://download.pytorch.org/whl/nightly/cu124"])
228
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "-r", "requirements.txt"])