Adamiros commited on
Commit
f280570
·
verified ·
1 Parent(s): b117132

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -304,7 +304,15 @@ with demo:
304
  with gr.Row():
305
  with gr.Column():
306
  model_name_textbox = gr.Textbox(label="Model name")
307
- precision = gr.Radio(["bfloat16", "float16", "4bit"], label="Precision", info="What precision are you using for inference?")
 
 
 
 
 
 
 
 
308
  hf_model_id = gr.Textbox(label="Huggingface Model ID")
309
  contact_email = gr.Textbox(label="E-Mail")
310
  file_output = gr.File()
 
304
  with gr.Row():
305
  with gr.Column():
306
  model_name_textbox = gr.Textbox(label="Model name")
307
+ #precision = gr.Radio(["bfloat16", "float16", "4bit"], label="Precision", info="What precision are you using for inference?")
308
+ #precision_options = []
309
+ precision = gr.Dropdown(
310
+ choices=[i.value.name for i in Precision if i != Precision.Unknown],
311
+ label="Precision",
312
+ multiselect=False,
313
+ value="float16",
314
+ interactive=True,
315
+ )
316
  hf_model_id = gr.Textbox(label="Huggingface Model ID")
317
  contact_email = gr.Textbox(label="E-Mail")
318
  file_output = gr.File()