Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 1f92f7cc authored by Atream's avatar Atream
Browse files

[fix] linux and windows can all find CPUInfer in current Directory

parent 0e613b60
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
ktransformers/operators/experts.py
View file @ 1f92f7cc
......@@ -19,8 +19,9 @@ import torch
import sys, os
from ktransformers.operators.base_operator import BaseInjectedModule
#sys.path.append(os.path.dirname(__file__) + "/../ktransformers_ext/build/")
sys.path.append(os.path.dirname(__file__) + "\\..\\ktransformers_ext\\build\\Release")
sys.path.append(os.path.join(os.path.dirname(__file__), "..", "ktransformers_ext", "build"))
sys.path.append(os.path.join(os.path.dirname(__file__), "..", "ktransformers_ext", "build", "Release"))
sys.path.append(os.path.join(os.path.dirname(__file__), "..", "ktransformers_ext", "build", "Debug"))
import cpuinfer_ext
from cpuinfer_ext.moe import MOEConfig, MOE
import ctypes
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment