Upload config.yml with huggingface_hub
Browse files- config.yml +2 -1
config.yml
CHANGED
|
@@ -1,10 +1,11 @@
|
|
| 1 |
hparams:
|
|
|
|
| 2 |
nhead: 8
|
| 3 |
low_rank_dim: 64
|
| 4 |
query_residual_weight: 0.5
|
| 5 |
anchor_scale: 0.8
|
| 6 |
num_steps: 4
|
| 7 |
num_layers: 6
|
| 8 |
-
embedding_dim:
|
| 9 |
use_fro_norm: true
|
| 10 |
feature_extractor: openai/clip-vit-large-patch14
|
|
|
|
| 1 |
hparams:
|
| 2 |
+
hidden_dim: 512
|
| 3 |
nhead: 8
|
| 4 |
low_rank_dim: 64
|
| 5 |
query_residual_weight: 0.5
|
| 6 |
anchor_scale: 0.8
|
| 7 |
num_steps: 4
|
| 8 |
num_layers: 6
|
| 9 |
+
embedding_dim: 768
|
| 10 |
use_fro_norm: true
|
| 11 |
feature_extractor: openai/clip-vit-large-patch14
|