mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix dynatemp parameters always visible

Browse source
This commit is contained in:
oobabooga 2024-01-08 19:42:31 -08:00
parent 29c2693ea0
commit 372ef5e2d8
2 changed files with 14 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

18
modules/loaders.py
View file

@ -356,12 +356,20 @@ def list_all_samplers():
return sorted(all_samplers)
def blacklist_samplers(loader):
def blacklist_samplers(loader, dynamic_temperature):
all_samplers = list_all_samplers()
if loader == 'All':
return [gr.update(visible=True) for sampler in all_samplers]
else:
return [gr.update(visible=True) if sampler in loaders_samplers[loader] else gr.update(visible=False) for sampler in all_samplers]
output = []
for sampler in all_samplers:
if loader == 'All' or sampler in loaders_samplers[loader]:
if sampler.startswith('dynatemp'):
output.append(gr.update(visible=dynamic_temperature))
else:
output.append(gr.update(visible=True))
else:
output.append(gr.update(visible=False))
return output
def get_model_types(loader):