USA-Coin-Retrieval / local_config.yaml
breezedeus's picture
update
5315596
raw
history blame
438 Bytes
dataset:
data_path: "data/coin_usa"
detector:
model_name: "google/owlvit-base-patch32"
device: "cpu"
resized_to: 300
coin_clip_db:
db_dir: "data/coin_clip_chroma.db"
model_name: "../coin-clip-vit-base-patch32"
collection_name: "coin_clip_collection"
device: "cpu"
clip_db:
db_dir: "data/clip_chroma.db"
model_name: "openai/clip-vit-base-patch32"
collection_name: "clip_collection"
device: "cpu"
device: "cpu"