USA-Coin-Retrieval / hf_config.yaml
breezedeus's picture
update
5315596
raw
history blame contribute delete
453 Bytes
dataset:
data_path: "breezedeus/usa-coins"
detector:
model_name: "google/owlvit-base-patch32"
device: "cpu"
resized_to: 300
coin_clip_db:
db_dir: "data/coin_clip_chroma.db"
model_name: "breezedeus/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"