Skip to content

GitLab

Commit d76078ef authored by yangql's avatar yangql
Browse files

initing develop

parent a2630e0f
Hide whitespace changes
Inline Side-by-side
Showing with 20 additions and 20 deletions
auto_gptq/nn_modules/qlinear/qlinear_cuda.py
View file @ d76078ef
......@@ -9,16 +9,16 @@ import transformers
logger = getLogger(__name__)
try:
import autogptq_cuda_64
import autogptq_cuda_256
_autogptq_cuda_available = True
except ImportError:
logger.warning("CUDA extension not installed.")
autogptq_cuda_256 = None
autogptq_cuda_64 = None
_autogptq_cuda_available = False
# try:
# import autogptq_cuda_64
# import autogptq_cuda_256
# _autogptq_cuda_available = True
# except ImportError:
# logger.warning("CUDA extension not installed.")
autogptq_cuda_256 = None
autogptq_cuda_64 = None
_autogptq_cuda_available = False
class QuantLinear(nn.Module):
......
auto_gptq/nn_modules/qlinear/qlinear_cuda_old.py
View file @ d76078ef
......@@ -8,16 +8,16 @@ import transformers
logger = getLogger(__name__)
try:
import autogptq_cuda_64
import autogptq_cuda_256
_autogptq_cuda_available = True
except ImportError:
logger.warning("CUDA extension not installed.")
autogptq_cuda_256 = None
autogptq_cuda_64 = None
_autogptq_cuda_available = False
# try:
# import autogptq_cuda_64
# import autogptq_cuda_256
# _autogptq_cuda_available = True
# except ImportError:
# logger.warning("CUDA extension not installed.")
autogptq_cuda_256 = None
autogptq_cuda_64 = None
_autogptq_cuda_available = False
class QuantLinear(nn.Module):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment