Spaces:
Runtime error
Runtime error
bugfix on n_threads default
Browse files
app.py
CHANGED
|
@@ -193,7 +193,7 @@ def main():
|
|
| 193 |
logging.basicConfig(level=logging.INFO)
|
| 194 |
default = '256M'
|
| 195 |
default_verbose = True
|
| 196 |
-
default_threads =
|
| 197 |
mf, cf = get_weight_files(default)
|
| 198 |
|
| 199 |
with gr.Blocks() as demo:
|
|
@@ -203,7 +203,7 @@ def main():
|
|
| 203 |
model_dd = gr.Dropdown(mf, value=mf[0], label='Decoder Weights')
|
| 204 |
clip_dd = gr.Dropdown(cf, value=cf[0], label='CLIP Weights')
|
| 205 |
verbose_cb= gr.Checkbox(value=default_verbose, label='Verbose Mode')
|
| 206 |
-
thread_dd = gr.Slider(minimum=1, maximum=
|
| 207 |
|
| 208 |
def on_size_change(sz, verbose, n_threads):
|
| 209 |
mlist, clist = get_weight_files(sz)
|
|
|
|
| 193 |
logging.basicConfig(level=logging.INFO)
|
| 194 |
default = '256M'
|
| 195 |
default_verbose = True
|
| 196 |
+
default_threads = 2
|
| 197 |
mf, cf = get_weight_files(default)
|
| 198 |
|
| 199 |
with gr.Blocks() as demo:
|
|
|
|
| 203 |
model_dd = gr.Dropdown(mf, value=mf[0], label='Decoder Weights')
|
| 204 |
clip_dd = gr.Dropdown(cf, value=cf[0], label='CLIP Weights')
|
| 205 |
verbose_cb= gr.Checkbox(value=default_verbose, label='Verbose Mode')
|
| 206 |
+
thread_dd = gr.Slider(minimum=1, maximum=os.cpu_count(), step=1, value=default_threads, label='CPU Threads (n_threads)')
|
| 207 |
|
| 208 |
def on_size_change(sz, verbose, n_threads):
|
| 209 |
mlist, clist = get_weight_files(sz)
|