chrisxx commited on
Commit
36ada0b
·
verified ·
1 Parent(s): 0ade316

Update app.py

Browse files

this is handled in the sampling function now

Files changed (1) hide show
  1. app.py +0 -3
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