squall/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Minor

Browse Source
This commit is contained in:
Woosuk Kwon 2023-02-22 18:08:25 +00:00
parent 709a69176e
commit 608f74ffe5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cacheflow/models/model_utils.py
View File

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