make transformers<5
Browse files
dreamlite/pipelines/dreamlite/pipeline_dreamlite.py
CHANGED
|
@@ -185,7 +185,7 @@ class DreamLitePipeline(
|
|
| 185 |
attention_mask=tk_out.attention_mask,
|
| 186 |
pixel_values=tk_out.pixel_values,
|
| 187 |
image_grid_thw=tk_out.image_grid_thw,
|
| 188 |
-
mm_token_type_ids=tk_out.mm_token_type_ids,
|
| 189 |
output_hidden_states=True
|
| 190 |
)
|
| 191 |
|
|
|
|
| 185 |
attention_mask=tk_out.attention_mask,
|
| 186 |
pixel_values=tk_out.pixel_values,
|
| 187 |
image_grid_thw=tk_out.image_grid_thw,
|
| 188 |
+
# mm_token_type_ids=tk_out.mm_token_type_ids,
|
| 189 |
output_hidden_states=True
|
| 190 |
)
|
| 191 |
|
dreamlite/pipelines/dreamlite/pipeline_dreamlite_mobile.py
CHANGED
|
@@ -163,7 +163,7 @@ class DreamLiteMobilePipeline(
|
|
| 163 |
attention_mask=tk_out.attention_mask,
|
| 164 |
pixel_values=tk_out.pixel_values,
|
| 165 |
image_grid_thw=tk_out.image_grid_thw,
|
| 166 |
-
mm_token_type_ids=tk_out.mm_token_type_ids,
|
| 167 |
output_hidden_states=True
|
| 168 |
)
|
| 169 |
|
|
|
|
| 163 |
attention_mask=tk_out.attention_mask,
|
| 164 |
pixel_values=tk_out.pixel_values,
|
| 165 |
image_grid_thw=tk_out.image_grid_thw,
|
| 166 |
+
# mm_token_type_ids=tk_out.mm_token_type_ids,
|
| 167 |
output_hidden_states=True
|
| 168 |
)
|
| 169 |
|
requirements.txt
CHANGED
|
@@ -2,7 +2,7 @@ torch
|
|
| 2 |
torchvision
|
| 3 |
numpy
|
| 4 |
Pillow
|
| 5 |
-
transformers
|
| 6 |
diffusers
|
| 7 |
accelerate
|
| 8 |
gradio
|
|
|
|
| 2 |
torchvision
|
| 3 |
numpy
|
| 4 |
Pillow
|
| 5 |
+
transformers==4.57.3
|
| 6 |
diffusers
|
| 7 |
accelerate
|
| 8 |
gradio
|