Update lvd_pipeline.py
Browse files- lvd_pipeline.py +1 -1
lvd_pipeline.py
CHANGED
|
@@ -742,7 +742,7 @@ class GroundedTextToVideoSDPipeline(DiffusionPipeline, TextualInversionLoaderMix
|
|
| 742 |
lvd_gligen_phrases_frame = lvd_gligen_phrases_frame[:max_objs]
|
| 743 |
lvd_gligen_boxes_frame = lvd_gligen_boxes_frame[:max_objs]
|
| 744 |
|
| 745 |
-
|
| 746 |
|
| 747 |
if n_objs:
|
| 748 |
# prepare batched input to the PositionNet (boxes, phrases, mask)
|
|
|
|
| 742 |
lvd_gligen_phrases_frame = lvd_gligen_phrases_frame[:max_objs]
|
| 743 |
lvd_gligen_boxes_frame = lvd_gligen_boxes_frame[:max_objs]
|
| 744 |
|
| 745 |
+
n_objs = len(lvd_gligen_boxes_frame)
|
| 746 |
|
| 747 |
if n_objs:
|
| 748 |
# prepare batched input to the PositionNet (boxes, phrases, mask)
|