vine / config.json
ASethi04's picture
Initial upload of VINE model architecture and weights - model and weights
1472aed verified
{
"_device": "cuda",
"alpha": 0.5,
"architectures": [
"VineModel"
],
"auto_map": {
"AutoConfig": "vine_config.VineConfig",
"AutoModel": "vine_model.VineModel"
},
"bbox_min_dim": 5,
"box_threshold": 0.35,
"debug_visualizations": false,
"hidden_dim": 768,
"interested_object_pairs": [],
"local_dir": null,
"local_filename": null,
"max_video_length": 100,
"model_file": null,
"model_name": "openai/clip-vit-base-patch32",
"model_repo": "KevinX-Penn28/testing",
"model_type": "vine",
"multi_class": false,
"num_top_pairs": 18,
"output_logit": false,
"return_flattened_segments": false,
"return_valid_pairs": false,
"segmentation_method": "grounding_dino_sam2",
"target_fps": 1,
"text_threshold": 0.25,
"topk_cate": 3,
"torch_dtype": "float32",
"transformers_version": "4.46.2",
"use_hf_repo": true,
"visualization_dir": null,
"visualize": false,
"white_alpha": 0.8
}