Fix DiffusionOutput return statement in LTX2Pipeline
Co-authored-by: hsliuustc0106 <222337142+hsliuustc0106@users.noreply.github.com>
This commit is contained in:
1 parent
99c02d3871
commit
d3a13425d9
1 file changed
+1
-4
@@ -190,7 +190,4 @@ class LTX2Pipeline(nn.Module):
|
||||
frames = output
|
||||
|
||||
# Return as DiffusionOutput
|
||||
return DiffusionOutput(
|
||||
images=frames,
|
||||
final_output_type="image",
|
||||
)
|
||||
return DiffusionOutput(output=frames)
|
||||
Reference in new issue
Block a user