mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-01-23 15:24:28 +08:00
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…
x
Reference in New Issue
Block a user