瀏覽代碼

add img embedding.py

luojunhui 4 周之前
父節點
當前提交
3e86a5f79d
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      applications/clip_embedding/embedding.py

+ 1 - 1
applications/clip_embedding/embedding.py

@@ -8,7 +8,7 @@ from PIL import Image
 from .clip_model import get_model
 from .clip_model import get_model
 
 
 # init 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:
 def _normalize(x: torch.Tensor) -> torch.Tensor: