rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Fix for multi-gpu not showing appropriately 

Slight visual improvement for custom model load breadcrumbs
This commit is contained in:
ebolam 2022-06-08 19:39:04 -04:00
parent 4548dcf1b0
commit c565978fff
3 changed files with 30 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
aiserver.py
View File

@ -1031,6 +1031,7 @@ def get_model_info(model, directory=""):
key_value = "" key_value = ""
break_values = [] break_values = []
url = False url = False
gpu_count = torch.cuda.device_count()
if model in [x[1] for x in model_menu['apilist']]: if model in [x[1] for x in model_menu['apilist']]:
if path.exists("settings/{}.settings".format(model)): if path.exists("settings/{}.settings".format(model)):
with open("settings/{}.settings".format(model), "r") as file: with open("settings/{}.settings".format(model), "r") as file:
@ -1059,10 +1060,10 @@ def get_model_info(model, directory=""):
break_values = file.read().split(",") break_values = file.read().split(",")
else: else:
break_values = [layer_count] break_values = [layer_count]
break_values += [0] * (gpu+1 - len(break_values)) break_values += [0] * (gpu_count - len(break_values))
emit('from_server', {'cmd': 'selected_model_info', 'key_value': key_value, 'key':key, emit('from_server', {'cmd': 'selected_model_info', 'key_value': key_value, 'key':key,
'gpu':gpu, 'layer_count':layer_count, 'breakmodel':breakmodel, 'gpu':gpu, 'layer_count':layer_count, 'breakmodel':breakmodel,
'break_values': break_values, 'gpu_count': torch.cuda.device_count(), 'break_values': break_values, 'gpu_count': gpu_count,
'url': url}, broadcast=True) 'url': url}, broadcast=True)
if key_value != "": if key_value != "":
get_oai_models(key_value) get_oai_models(key_value)

5
static/application.js
View File

@ -997,7 +997,7 @@ function buildLoadModelList(ar, menu, breadcrumbs) {
$("#loadmodellistbreadcrumbs").html(""); $("#loadmodellistbreadcrumbs").html("");
var i; var i;
for(i=0; i<breadcrumbs.length; i++) { for(i=0; i<breadcrumbs.length; i++) {
$("#loadmodellistbreadcrumbs").append("<button id='model_breadcrumbs"+i+"' name='"+ar[0][1]+"' value='"+breadcrumbs[i][0]+"'>"+breadcrumbs[i][1]+"</button>"); $("#loadmodellistbreadcrumbs").append("<button class=\"breadcrumbitem\" id='model_breadcrumbs"+i+"' name='"+ar[0][1]+"' value='"+breadcrumbs[i][0]+"'>"+breadcrumbs[i][1]+"</button><font color=white>\\</font>");
$("#model_breadcrumbs"+i).off("click").on("click", (function () { $("#model_breadcrumbs"+i).off("click").on("click", (function () {
return function () { return function () {
socket.send({'cmd': 'selectmodel', 'data': $(this).attr("name"), 'folder': $(this).attr("value")}); socket.send({'cmd': 'selectmodel', 'data': $(this).attr("name"), 'folder': $(this).attr("value")});
@ -1005,6 +1005,9 @@ function buildLoadModelList(ar, menu, breadcrumbs) {
} }
})(i)); })(i));
} }
if (breadcrumbs.length > 0) {
$("#loadmodellistbreadcrumbs").append("<hr size='1'>")
}
for(i=0; i<ar.length; i++) { for(i=0; i<ar.length; i++) {
var html var html
html = "<div class=\"flex\">\ html = "<div class=\"flex\">\

24
static/custom.css
View File

@ -1035,7 +1035,7 @@ body.connected .statusiconlabel, .statusiconlabel.always-available {
} }
.loadlistitem { .loadlistitem {
padding: 5px 10px 5px 10px; padding: 0px 0px 0px 0px;
display: flex; display: flex;
flex-grow: 1; flex-grow: 1;
color: #ffffff; color: #ffffff;
@ -1051,6 +1051,28 @@ body.connected .statusiconlabel, .statusiconlabel.always-available {
background-color: #688f1f; background-color: #688f1f;
} }
.breadcrumbitem {
padding: 5px 10px 5px 10px;
color: #ffffff;
background-color: transparent;
border: none;
-moz-transition: background-color 0.25s ease-in;
-o-transition: background-color 0.25s ease-in;
-webkit-transition: background-color 0.25s ease-in;
transition: background-color 0.25s ease-in;
}
.breadcrumbitem:hover {
cursor: pointer;
background-color: #688f1f;
}
hr {
padding: 0px;
margin: 0px;
}
.loadlistpadding { .loadlistpadding {
padding-right: 10px; padding-right: 10px;
} }