CARLEXsX commited on
Commit
31fbf3b
·
verified ·
1 Parent(s): d4ee4b6

Update ltx_video/models/transformers/transformer3d.py

Browse files
ltx_video/models/transformers/transformer3d.py CHANGED
@@ -1,4 +1,4 @@
1
- --- START OF MODIFIED FILE app_fluxContext_Ltx/ltx_video/models/transformers/transformer3d.py ---
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)