multimodalart HF Staff commited on
Commit
22d199e
·
verified ·
1 Parent(s): f2b3fb0

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -238,7 +238,7 @@ def stream(session_id: str) -> str:
238
  cond = model._conditioning((list(tokens) + [-1] * model.num_musiccoca)[:model.num_musiccoca],
239
  nvec, drm, cfgs)
240
  source = model.model.encode(cond).to(dt)
241
- sampler = make_sampler(c.get("temperature", 1.3), c.get("top_k", 40), gen)
242
  toks.append(model.model.decoder.step_f(dstate, source, sampler=sampler,
243
  temporal_step=model._temporal_step, depth_step=model._depth_step))
244
  history = torch.cat([history] + toks, dim=1)
 
238
  cond = model._conditioning((list(tokens) + [-1] * model.num_musiccoca)[:model.num_musiccoca],
239
  nvec, drm, cfgs)
240
  source = model.model.encode(cond).to(dt)
241
+ sampler = make_sampler(c.get("temperature", 1.1), c.get("top_k", 50), gen)
242
  toks.append(model.model.decoder.step_f(dstate, source, sampler=sampler,
243
  temporal_step=model._temporal_step, depth_step=model._depth_step))
244
  history = torch.cat([history] + toks, dim=1)