Procházet zdrojové kódy

add img embedding.py

luojunhui před 4 týdny
rodič
revize
315abc9cb9
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      applications/clip_embedding/clip_model.py

+ 1 - 1
applications/clip_embedding/clip_model.py

@@ -3,7 +3,7 @@ import torch
 
 from transformers import AutoModel, AutoConfig, AutoTokenizer
 
-MODEL_NAME = "BAAI/EVA-CLIP-8B-plus"
+MODEL_NAME = "BAAI/EVA-CLIP-8B-448"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 MAX_BATCH = int(os.getenv("MAX_BATCH", "32"))