|
@@ -8,7 +8,7 @@ from PIL import Image
|
|
|
from .clip_model import get_model
|
|
|
|
|
|
# init model
|
|
|
-model, processor, DEVICE, DTYPE, MAX_BATCH = get_model()
|
|
|
+model, image_processor, tokenizer, DEVICE, DTYPE, MAX_BATCH = get_model()
|
|
|
|
|
|
|
|
|
def _normalize(x: torch.Tensor) -> torch.Tensor:
|