Correct issues caused by merging main

This commit is contained in:
Brandon Rising 2023-07-26 12:25:46 -04:00
parent c16da75ac7
commit 861c0fe76b
13 changed files with 613 additions and 601 deletions

View File

@ -583,7 +583,10 @@ class IAIOnnxRuntimeModel:
providers.append(self.provider) providers.append(self.provider)
else: else:
providers = get_available_providers() providers = get_available_providers()
self.session = InferenceSession(self.proto.SerializeToString(), providers=providers, sess_options=sess) try:
self.session = InferenceSession(self.proto.SerializeToString(), providers=providers, sess_options=sess)
except Exception as e:
raise e
#self.session = InferenceSession("tmp.onnx", providers=[self.provider], sess_options=self.sess_options) #self.session = InferenceSession("tmp.onnx", providers=[self.provider], sess_options=self.sess_options)
# self.io_binding = self.session.io_binding() # self.io_binding = self.session.io_binding()

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

File diff suppressed because one or more lines are too long

View File

@ -12,7 +12,7 @@
margin: 0; margin: 0;
} }
</style> </style>
<script type="module" crossorigin src="./assets/index-e45bf5a6.js"></script> <script type="module" crossorigin src="./assets/index-efb4c895.js"></script>
</head> </head>
<body dir="ltr"> <body dir="ltr">

View File

@ -23,6 +23,7 @@
"menu": "Menu" "menu": "Menu"
}, },
"common": { "common": {
"communityLabel": "Community",
"hotkeysLabel": "Hotkeys", "hotkeysLabel": "Hotkeys",
"darkMode": "Dark Mode", "darkMode": "Dark Mode",
"lightMode": "Light Mode", "lightMode": "Light Mode",

View File

@ -8,7 +8,7 @@ export const modelIdToMainModelParam = (
mainModelId: string mainModelId: string
): MainModelParam | undefined => { ): MainModelParam | undefined => {
const log = logger('models'); const log = logger('models');
const [base_model, _model_type, model_name] = mainModelId.split('/'); const [base_model, model_type, model_name] = mainModelId.split('/');
const result = zMainModel.safeParse({ const result = zMainModel.safeParse({
base_model, base_model,

View File

@ -153,7 +153,15 @@ const createModelEntities = <T extends AnyModelConfigEntity>(
export const modelsApi = api.injectEndpoints({ export const modelsApi = api.injectEndpoints({
endpoints: (build) => ({ endpoints: (build) => ({
getOnnxModels: build.query<EntityState<OnnxModelConfigEntity>, void>({ getOnnxModels: build.query<EntityState<OnnxModelConfigEntity>, void>({
query: () => ({ url: 'models/', params: { model_type: 'onnx' } }), query: (base_models) => {
const params = {
model_type: 'onnx',
base_models,
};
const query = queryString.stringify(params, { arrayFormat: 'none' });
return `models/?${query}`;
},
providesTags: (result, error, arg) => { providesTags: (result, error, arg) => {
const tags: ApiFullTagDescription[] = [ const tags: ApiFullTagDescription[] = [
{ id: 'OnnxModel', type: LIST_TAG }, { id: 'OnnxModel', type: LIST_TAG },