Remove unused imports.
parent
42b06c8bfc
commit
58eab31083
2
model.py
2
model.py
|
@ -1,6 +1,6 @@
|
||||||
import torch
|
import torch
|
||||||
from tensorizer import TensorDeserializer
|
from tensorizer import TensorDeserializer
|
||||||
from tensorizer.utils import no_init_or_tensor, convert_bytes, get_mem_usage
|
from tensorizer.utils import no_init_or_tensor
|
||||||
from transformers import (
|
from transformers import (
|
||||||
GPTNeoXTokenizerFast,
|
GPTNeoXTokenizerFast,
|
||||||
LogitsProcessor,
|
LogitsProcessor,
|
||||||
|
|
Loading…
Reference in New Issue