Spaces:
Running
on
Zero
Running
on
Zero
eval
Browse files
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.
|
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)
|