Merge branch 'main' into build-utils-cleanup

Author: swolchok

install/install_requirements.sh

@@ -53,7 +53,7 @@ PYTORCH_NIGHTLY_VERSION=dev20241002
 VISION_NIGHTLY_VERSION=dev20241002
 
 # Nightly version for torchtune
-TUNE_NIGHTLY_VERSION=dev20240928
+TUNE_NIGHTLY_VERSION=dev20241010
 
 # Uninstall triton, as nightly will depend on pytorch-triton, which is one and the same
 (
@@ -78,7 +78,7 @@ fi
 REQUIREMENTS_TO_INSTALL=(
   torch=="2.6.0.${PYTORCH_NIGHTLY_VERSION}"
   torchvision=="0.20.0.${VISION_NIGHTLY_VERSION}"
-  torchtune=="0.3.0.${TUNE_NIGHTLY_VERSION}"
+  torchtune=="0.4.0.${TUNE_NIGHTLY_VERSION}"
 )
 
 # Install the requirements. --extra-index-url tells pip to look for package

install/requirements.txt

@@ -12,7 +12,8 @@ tiktoken
 # Miscellaneous
 snakeviz
 sentencepiece
-numpy
+# numpy version range required by GGUF util
+numpy >= 1.17, < 2.0
 gguf
 blobfile
 tomli >= 1.1.0 ; python_version < "3.11"

torchchat/model.py

@@ -12,8 +12,6 @@
 from enum import Enum
 from pathlib import Path
 
-import torchvision
-
 from typing import Any, Callable, Dict, Optional, Union
 from collections.abc import Hashable