Luigi commited on
Commit
aa69ba7
·
1 Parent(s): 23c5da5

bugfix on n_threads default

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = os.cpu_count() or 1
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=default_threads, step=1, value=2, label='CPU Threads (n_threads)')
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)