joelniklaus HF Staff commited on
Commit
0b70925
·
1 Parent(s): 18b5be6

add arrow back

Browse files
app/src/content/embeds/d3-pipeline.html CHANGED
@@ -101,6 +101,7 @@
101
  { from: 'transform', to: 'write' },
102
  { from: 'transform', to: 'rollout' },
103
  { from: 'rollout', to: 'vllm' },
 
104
  { from: 'write', to: 'hf_out' },
105
  { from: 'write', to: 'card' },
106
  { from: 'write', to: 'monitor' },
@@ -256,6 +257,11 @@
256
  if (e.from === 'transform' && e.to === 'write') return vBez(pt(f,'bottom'), pt(t,'top'));
257
  if (e.from === 'transform' && e.to === 'rollout') return hBez(pt(f,'left'), pt(t,'right'));
258
  if (e.from === 'rollout' && e.to === 'vllm') return vBez(pt(f,'bottom'), pt(t,'top'));
 
 
 
 
 
259
 
260
  // Fan out from Write: top/center/bottom of right edge
261
  const sp = Math.round(f._h * 0.28);
 
101
  { from: 'transform', to: 'write' },
102
  { from: 'transform', to: 'rollout' },
103
  { from: 'rollout', to: 'vllm' },
104
+ { from: 'vllm', to: 'transform', label: 'response' },
105
  { from: 'write', to: 'hf_out' },
106
  { from: 'write', to: 'card' },
107
  { from: 'write', to: 'monitor' },
 
257
  if (e.from === 'transform' && e.to === 'write') return vBez(pt(f,'bottom'), pt(t,'top'));
258
  if (e.from === 'transform' && e.to === 'rollout') return hBez(pt(f,'left'), pt(t,'right'));
259
  if (e.from === 'rollout' && e.to === 'vllm') return vBez(pt(f,'bottom'), pt(t,'top'));
260
+ if (e.from === 'vllm' && e.to === 'transform') {
261
+ const a = pt(f, 'right');
262
+ const b = pt(t, 'left', Math.round(t._h * 0.2));
263
+ return hBez(a, b);
264
+ }
265
 
266
  // Fan out from Write: top/center/bottom of right edge
267
  const sp = Math.round(f._h * 0.28);