Minor

2023-02-22 18:08:25 +00:00
parent 709a69176e
commit 608f74ffe5
1 changed files with 1 additions and 1 deletions
--- 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,