diff --git a/.github/workflows/regression_test.yml b/.github/workflows/regression_test.yml index 971515747e..4b7877ea91 100644 --- a/.github/workflows/regression_test.yml +++ b/.github/workflows/regression_test.yml @@ -58,7 +58,7 @@ jobs: gpu-arch-type: ${{ matrix.gpu-arch-type }} gpu-arch-version: ${{ matrix.gpu-arch-version }} script: | - conda create -n venv python=3.9 -y + conda create -n venv python=3.8 -y conda activate venv echo "::group::Install newer objcopy that supports --set-section-alignment" yum install -y devtoolset-10-binutils diff --git a/torchao/quantization/fp6_llm.py b/torchao/quantization/fp6_llm.py index 0fb0f7dd98..0c405b8235 100644 --- a/torchao/quantization/fp6_llm.py +++ b/torchao/quantization/fp6_llm.py @@ -1,5 +1,5 @@ import math -from typing import Optional +from typing import List, Optional import torch from torch import nn, Tensor @@ -291,7 +291,7 @@ def extra_repr(self) -> str: return f'in_features={self.in_features}, out_features={self.out_features}, bias={self.bias is not None}' -def convert_fp6_llm(model: nn.Module, skip_fqn_list: Optional[list[str]] = None, cur_fqn: str = "") -> None: +def convert_fp6_llm(model: nn.Module, skip_fqn_list: Optional[List[str]] = None, cur_fqn: str = "") -> None: for name, child in model.named_children(): new_fqn = name if cur_fqn == "" else f"{cur_fqn}.{name}"