(inputs)
| 60 | diffusion_model = model_patcher.model.diffusion_model |
| 61 | |
| 62 | def call_model(inputs): |
| 63 | x, t, context, attention_mask = inputs |
| 64 | return diffusion_model(x, t, context=context, attention_mask=None) |
| 65 | |
| 66 | # scheduler |
| 67 | scheduler = diffusers.FlowMatchEulerDiscreteScheduler(shift=sample_shift) |