Update app.py
Browse filesthis is handled in the sampling function now
app.py
CHANGED
|
@@ -208,9 +208,6 @@ def initialize_model():
|
|
| 208 |
# Sample with the fresh noise (matching notebook: sample(model, noise, actions[:, aidx:aidx+1], ...))
|
| 209 |
z = sample(model_, noise, action_buf, num_steps=n_steps, cfg=cfg, negative_actions=None)
|
| 210 |
|
| 211 |
-
# Update cache location after sample (matching notebook: model.cache.update_global_location(1))
|
| 212 |
-
cache.update_global_location(1)
|
| 213 |
-
|
| 214 |
if clamp:
|
| 215 |
z = t.clamp(z, -1, 1)
|
| 216 |
return z
|
|
|
|
| 208 |
# Sample with the fresh noise (matching notebook: sample(model, noise, actions[:, aidx:aidx+1], ...))
|
| 209 |
z = sample(model_, noise, action_buf, num_steps=n_steps, cfg=cfg, negative_actions=None)
|
| 210 |
|
|
|
|
|
|
|
|
|
|
| 211 |
if clamp:
|
| 212 |
z = t.clamp(z, -1, 1)
|
| 213 |
return z
|