openbmb

/

MiniCPM-V-2_6

like 806

Follow

OpenBMB 348

 6 import torch.nn as nn
  8 # isort: off
  9 # We need to import the CUDA kernels after importing torch

---> 10 import flash_attn_2_cuda as flash_attn_cuda
12 # isort: on
14 def maybe_contiguous(x):

ImportError: libcudart.so.11.0: cannot open shared object file: No such file or directory