From 608f74ffe5635cd8ac7fc78a13e3d67a4f59b9e0 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Wed, 22 Feb 2023 18:08:25 +0000 Subject: [PATCH] Minor --- cacheflow/models/model_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cacheflow/models/model_utils.py b/cacheflow/models/model_utils.py index 3878b87e..522630d7 100644 --- a/cacheflow/models/model_utils.py +++ b/cacheflow/models/model_utils.py @@ -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,