Spaces:
Running
Running
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)
|