Upgrading to python3.13? [testing needed]

setup.py

@@ -26,7 +26,7 @@ def get_system():
     "toml",
     "GitPython",
     "colorlog",
-    "cocotb[bus]==1.8.0",
+    "cocotb[bus]",
     "pytest",
     "pytorch-lightning",
     "transformers",
@@ -37,7 +37,6 @@ def get_system():
     "evaluate",
     "ipython",
     "ipdb",
-    "sentencepiece",
     "einops",
     "pybind11",
     "tabulate",

src/chop/distributed/tensor/ops/utils.py

@@ -87,7 +87,7 @@ def wrapper(impl):
 
 
 def as_list(
-    x: Union[List[object], object]
+    x: Union[List[object], object],
     # pyre-fixme[11]: Annotation `immutable_list` is not defined as a type.
 ) -> Union[List[object], torch.fx.immutable_collections.immutable_list]:  # type: ignore[valid-type]
     # During tracing, `aten.sum.dim_IntList` uses `immutable_list` for its args,

src/chop/ir/graph/mase_graph.py

@@ -377,7 +377,7 @@ def from_checkpoint(
             MaseGraph: Loaded MaseGraph.
         """
         with open(f"{checkpoint}.pt", "rb") as f:
-            loaded_model = torch.load(f)
+            loaded_model = torch.load(f, weights_only=False)
 
         assert isinstance(
             loaded_model, fx.GraphModule

src/chop/models/bert/configuration_bert.py

@@ -13,7 +13,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" BERT model configuration"""
+"""BERT model configuration"""
 from collections import OrderedDict
 from typing import Mapping
 

src/chop/models/bert/configuration_bert_quantized.py

@@ -13,7 +13,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" BERT model configuration"""
+"""BERT model configuration"""
 from collections import OrderedDict
 from typing import Mapping
 

src/chop/models/llama/configuration_llama_llora.py

@@ -17,7 +17,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" LLaMA model configuration"""
+"""LLaMA model configuration"""
 
 from transformers.configuration_utils import PretrainedConfig
 from transformers.utils import logging

src/chop/models/llama/configuration_llama_quantized.py

@@ -17,7 +17,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" LLaMA model configuration"""
+"""LLaMA model configuration"""
 
 from transformers.configuration_utils import PretrainedConfig
 from transformers.utils import logging

src/chop/models/llama/configuration_llama_sparse.py

@@ -17,7 +17,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" LLaMA model configuration"""
+"""LLaMA model configuration"""
 
 from transformers.configuration_utils import PretrainedConfig
 from transformers.utils import logging

src/chop/models/llama/modeling_llama_llora.py

@@ -19,7 +19,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" PyTorch LLaMA model."""
+"""PyTorch LLaMA model."""
 
 import math
 from typing import List, Optional, Tuple, Union

src/chop/models/llama/modeling_llama_quantized.py

@@ -19,7 +19,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" PyTorch LLaMA model."""
+"""PyTorch LLaMA model."""
 
 import math
 from typing import List, Optional, Tuple, Union

src/chop/models/llama/modeling_llama_sparse.py

@@ -19,7 +19,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" PyTorch LLaMA model."""
+"""PyTorch LLaMA model."""
 
 import math
 from typing import List, Optional, Tuple, Union

src/chop/models/opt/configuration_opt.py

@@ -15,7 +15,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" OPT model configuration"""
+"""OPT model configuration"""
 from transformers.configuration_utils import PretrainedConfig
 from transformers.utils import logging
 

src/chop/models/opt/configuration_opt_lora.py

@@ -15,7 +15,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" OPT model configuration"""
+"""OPT model configuration"""
 from transformers.configuration_utils import PretrainedConfig
 from transformers.utils import logging
 from .lora_config_opt import parse_opt_lora_config

src/chop/models/opt/configuration_opt_quantized.py

@@ -12,7 +12,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" OPT model configuration"""
+"""OPT model configuration"""
 from transformers.configuration_utils import PretrainedConfig
 from .quant_config_opt import parse_opt_quantized_config
 

src/chop/models/opt/configuration_opt_sparse.py

@@ -15,7 +15,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" OPT model configuration"""
+"""OPT model configuration"""
 from transformers.configuration_utils import PretrainedConfig
 from transformers.utils import logging
 from .sparse_config_opt import parse_opt_sparse_config

src/chop/models/opt/modeling_opt.py

@@ -12,7 +12,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" PyTorch OPT model."""
+"""PyTorch OPT model."""
 from typing import List, Optional, Tuple, Union
 
 import torch

src/chop/models/opt/modeling_opt_lora.py

@@ -16,7 +16,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" PyTorch OPT model."""
+"""PyTorch OPT model."""
 import random
 from typing import Optional, Tuple, Union
 

src/chop/models/opt/modeling_opt_quantized.py

@@ -12,7 +12,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" PyTorch OPT model."""
+"""PyTorch OPT model."""
 import random
 from typing import List, Optional, Tuple, Union
 

src/chop/models/opt/modeling_opt_sparse.py

@@ -16,7 +16,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-""" PyTorch OPT model."""
+"""PyTorch OPT model."""
 import random
 from typing import Optional, Tuple, Union