YiYiXu HF Staff commited on
Commit
0bfe4c0
·
verified ·
1 Parent(s): c7967b3

Pin diffusers to abot-world-modular@206121e08 (tiny VAE decode in the streaming preset)

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@
2
  ABot-World on Modular Diffusers — interactive action-conditioned world rollout.
3
  gradio.Server + WebSocket live-backend edition.
4
 
5
- The engine is one `pipe.stream(action_source=...)` loop over the Modular Diffusers
6
  `ABotWorldStreamingBlocks` preset (huggingface/diffusers#14159): the pipeline polls the
7
  held keys once per generated block and yields every block's decoded frames.
8
  The serving shell (gradio.Server, WebSocket frame stream, pacing, per-session queues,
@@ -344,7 +344,7 @@ def create_gpu_rollout_loop(command_queue, seed_path, prompt_text, seed):
344
  image=image,
345
  height=STREAM_HEIGHT,
346
  width=STREAM_WIDTH,
347
- action_source=action_source,
348
  generator=torch.Generator("cpu").manual_seed(int(seed)),
349
  )
350
  for event in events:
 
2
  ABot-World on Modular Diffusers — interactive action-conditioned world rollout.
3
  gradio.Server + WebSocket live-backend edition.
4
 
5
+ The engine is one `pipe.stream(actions=<callable>)` loop over the Modular Diffusers
6
  `ABotWorldStreamingBlocks` preset (huggingface/diffusers#14159): the pipeline polls the
7
  held keys once per generated block and yields every block's decoded frames.
8
  The serving shell (gradio.Server, WebSocket frame stream, pacing, per-session queues,
 
344
  image=image,
345
  height=STREAM_HEIGHT,
346
  width=STREAM_WIDTH,
347
+ actions=action_source,
348
  generator=torch.Generator("cpu").manual_seed(int(seed)),
349
  )
350
  for event in events:
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  accelerate
2
- git+https://github.com/huggingface/diffusers.git@206121e08010ddc3451c3ad232ec166657a6da7f
3
  torch
4
  transformers
5
  ftfy
 
1
  accelerate
2
+ git+https://github.com/huggingface/diffusers.git@d2d0aeab86fd6e17c97d04266ecad5f25c168c24
3
  torch
4
  transformers
5
  ftfy