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
from text_generation_server.utils.gptq.quant_linear import QuantLinear
from loguru import logger
from typings import Optional
from typing import Optional
DEV = torch.device("cuda:0")