Merge branch 'main' into improve_fp6_llm_linear

pytorch · Jun 10, 2024 · 156063a · 156063a
2 parents fa38572 + 0d44041
commit 156063a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -1,5 +1,5 @@
 import math
-from typing import Optional, Tuple, List
+from typing import List, Optional, Tuple
 
 import torch
 from torch import nn, Tensor