MCPcopy Create free account
hub / github.com/dromara/Omega-AI / sample

Method sample

src/main/java/com/omega/example/dit/models/ICPlan.java:105–131  ·  view source on GitHub ↗

sample @param y0 is noise

(MMDiT_RoPE dit,Tensor y0, Tensor t, Tensor context, Tensor cos, Tensor sin, Tensor y1)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 7

test_ropeMethod · 0.95
test_rope2Method · 0.95
test_rope3Method · 0.95
test_rope4Method · 0.95
mmdit_rope_icplan_vavae_testMethod · 0.95
testODEMethod · 0.95
mmdit_rope_icplan_vavae_test2Method · 0.95

Calls 8

ininTMethod · 0.95
valMethod · 0.95
linear_interpMethod · 0.95
hostToDeviceMethod · 0.80
forwardMethod · 0.45
mulMethod · 0.45
addMethod · 0.45
copyGPUMethod · 0.45

Tested by 7

test_ropeMethod · 0.76
test_rope2Method · 0.76
test_rope3Method · 0.76
test_rope4Method · 0.76
mmdit_rope_icplan_vavae_testMethod · 0.76
testODEMethod · 0.76
mmdit_rope_icplan_vavae_test2Method · 0.76