This commit is contained in:
Woosuk Kwon 2023-02-22 18:08:25 +00:00
parent 709a69176e
commit 608f74ffe5

View File

@ -1,6 +1,6 @@
import torch.nn as nn
from cacheflow.worker.models.opt import OPTForCausalLM
from cacheflow.models.opt import OPTForCausalLM
MODEL_CLASSES = {
'opt': OPTForCausalLM,