Spaces:
Sleeping
Sleeping
Update ltx_video/models/transformers/transformer3d.py
Browse files
ltx_video/models/transformers/transformer3d.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
|
| 2 |
# Adapted from: https://github.com/huggingface/diffusers/blob/v0.26.3/src/diffusers/models/transformers/transformer_2d.py
|
| 3 |
import math
|
| 4 |
from dataclasses import dataclass
|
|
@@ -474,5 +474,4 @@ class Transformer3DModel(ModelMixin, ConfigMixin):
|
|
| 474 |
if not return_dict:
|
| 475 |
return (hidden_states,)
|
| 476 |
|
| 477 |
-
return Transformer3DModelOutput(sample=hidden_states)
|
| 478 |
-
--- END OF MODIFIED FILE app_fluxContext_Ltx/ltx_video/models/transformers/transformer3d.py ---
|
|
|
|
| 1 |
+
|
| 2 |
# Adapted from: https://github.com/huggingface/diffusers/blob/v0.26.3/src/diffusers/models/transformers/transformer_2d.py
|
| 3 |
import math
|
| 4 |
from dataclasses import dataclass
|
|
|
|
| 474 |
if not return_dict:
|
| 475 |
return (hidden_states,)
|
| 476 |
|
| 477 |
+
return Transformer3DModelOutput(sample=hidden_states)
|
|
|