mirror of
https://github.com/wassname/vllm.git
synced 2026-08-04 13:24:17 +08:00
Minor
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user