Skip to content

Commit

Permalink
Make quality and make style
Browse files Browse the repository at this point in the history
  • Loading branch information
TimoImhof committed Nov 14, 2023
1 parent 784770a commit 4eae90a
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions utils/back_comp/Utils.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import os
import random
import typing
from typing import Any, Union

import numpy as np
Expand Down Expand Up @@ -348,7 +347,7 @@ def generate_dummy_data(model: str = ""):
if model == "vit":
processor = ViTImageProcessor.from_pretrained("google/vit-base-patch16-224-in21k")
return processor(images=image, return_tensors="pt")

# For text just create and process a dummy string.
else:
input_ids = [i for i in range(20)]
Expand Down

0 comments on commit 4eae90a

Please sign in to comment.