Upload app.py with huggingface_hub
Browse files
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.
|
| 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)
|