Minor
This commit is contained in:
parent
709a69176e
commit
608f74ffe5
@ -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,
|
||||
|
||||
Loading…
Reference in New Issue
Block a user