This commit is contained in:
Nicolas Patry 2023-06-14 16:57:56 +02:00
parent 054a3d095c
commit 983c813f1d

View File

@ -15,7 +15,7 @@ import numpy as np
import torch import torch
from text_generation_server.utils.gptq.quant_linear import QuantLinear from text_generation_server.utils.gptq.quant_linear import QuantLinear
from loguru import logger from loguru import logger
from typings import Optional from typing import Optional
DEV = torch.device("cuda:0") DEV = torch.device("cuda:0")