Minor

608f74ff · Woosuk Kwon · 709a6917 · 608f74ff
Commit 608f74ff authored Feb 22, 2023 by Woosuk Kwon
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

cacheflow/models/model_utils.py cacheflow/models/model_utils.py +1 -1

No files found.
--- a/cacheflow/models/model_utils.py
+++ b/cacheflow/models/model_utils.py
 import torch.nn as nn

-from cacheflow.worker.models.opt import OPTForCausalLM
+from cacheflow.models.opt import OPTForCausalLM

 MODEL_CLASSES = {
    'opt': OPTForCausalLM,