skytnt commited on
Commit
94ac77e
1 Parent(s): b3f8835
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -385,7 +385,7 @@ if __name__ == "__main__":
385
  ckpt = torch.load(model_path, map_location="cpu", weights_only=True)
386
  state_dict = ckpt.get("state_dict", ckpt)
387
  model.load_state_dict(state_dict, strict=False)
388
- model.to(device="cpu", dtype=torch.float32)
389
  models[name] = model
390
 
391
  load_javascript()
@@ -477,7 +477,7 @@ if __name__ == "__main__":
477
  step=1, value=opt.max_gen // 2)
478
  with gr.Accordion("options", open=False):
479
  input_temp = gr.Slider(label="temperature", minimum=0.1, maximum=1.2, step=0.01, value=1)
480
- input_top_p = gr.Slider(label="top p", minimum=0.1, maximum=1, step=0.01, value=0.94)
481
  input_top_k = gr.Slider(label="top k", minimum=1, maximum=128, step=1, value=20)
482
  input_allow_cc = gr.Checkbox(label="allow midi cc event", value=True)
483
  input_render_audio = gr.Checkbox(label="render audio after generation", value=True)
 
385
  ckpt = torch.load(model_path, map_location="cpu", weights_only=True)
386
  state_dict = ckpt.get("state_dict", ckpt)
387
  model.load_state_dict(state_dict, strict=False)
388
+ model.to(device="cpu", dtype=torch.float32).eval()
389
  models[name] = model
390
 
391
  load_javascript()
 
477
  step=1, value=opt.max_gen // 2)
478
  with gr.Accordion("options", open=False):
479
  input_temp = gr.Slider(label="temperature", minimum=0.1, maximum=1.2, step=0.01, value=1)
480
+ input_top_p = gr.Slider(label="top p", minimum=0.1, maximum=1, step=0.01, value=0.95)
481
  input_top_k = gr.Slider(label="top k", minimum=1, maximum=128, step=1, value=20)
482
  input_allow_cc = gr.Checkbox(label="allow midi cc event", value=True)
483
  input_render_audio = gr.Checkbox(label="render audio after generation", value=True)