Skip to content

Commit 7146b33

Browse files
authored
Merge branch 'xieyongliang:master' into master
2 parents 2eaf31c + 3ca3a57 commit 7146b33

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

javascript/ui.js

+2
Original file line numberDiff line numberDiff line change
@@ -214,9 +214,11 @@ function restart_reload(){
214214

215215
function login(username) {
216216
var user=gradioApp().querySelector('#tabs').querySelectorAll('button')[5];
217+
var setting=gradioApp().querySelector('#tabs').querySelectorAll('button')[6];
217218

218219
if(username=='admin'){
219220
user.style.display='block'
221+
setting.style.display='none'
220222
user.click()
221223
}
222224
else

webui.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -234,7 +234,7 @@ def webui():
234234
api_endpoint = os.environ['api_endpoint']
235235
endpoint_name = os.environ['endpoint_name']
236236
for file in os.listdir(sd_models_dir):
237-
if os.path.isfile(os.path.join(sd_models_dir, file)) and (file.endswith('.ckpt') or file.endswith('.safesentors')):
237+
if os.path.isfile(os.path.join(sd_models_dir, file)) and (file.endswith('.ckpt') or file.endswith('.safetensors')):
238238
hash = modules.sd_models.model_hash(os.path.join(sd_models_dir, file))
239239
item = {}
240240
item['model_name'] = file
@@ -263,7 +263,7 @@ def webui():
263263
}
264264
for file in os.listdir(cn_models_dir):
265265
if os.path.isfile(os.path.join(cn_models_dir, file)) and \
266-
(file.endswith('pt') or file.endswith('.pth') or file.endswith('.ckpt') or file.endswith('.safetensors')):
266+
(file.endswith('.pt') or file.endswith('.pth') or file.endswith('.ckpt') or file.endswith('.safetensors')):
267267
hash = modules.sd_models.model_hash(os.path.join(cn_models_dir, file))
268268
item = {}
269269
item['model_name'] = file

0 commit comments

Comments
 (0)