Skip to content

Commit 8a56c17

Browse files
Add DinoV2 and update Keras hub version (#2165)
* add DinoV2 and update Keras hub version * Update scripts/hub_master.py Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> * pin TF versions * nit --------- Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 4cb3fe1 commit 8a56c17

File tree

3 files changed

+26
-3
lines changed

3 files changed

+26
-3
lines changed

requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,10 @@ pandas
1010
jupyter
1111
pydot
1212
boto3
13-
tensorflow
13+
tensorflow==2.19.1
1414
keras-cv
1515
keras-nlp
1616
keras-tuner
17-
tf-keras
17+
tf-keras==2.19.0
1818
keras-hub
1919
keras-rs-nightly

scripts/autogen.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
PROJECT_URL = {
3535
"keras": f"{KERAS_TEAM_GH}/keras/tree/v3.11.2/",
3636
"keras_tuner": f"{KERAS_TEAM_GH}/keras-tuner/tree/v1.4.7/",
37-
"keras_hub": f"{KERAS_TEAM_GH}/keras-hub/tree/v0.21.1/",
37+
"keras_hub": f"{KERAS_TEAM_GH}/keras-hub/tree/v0.22.1/",
3838
"tf_keras": f"{KERAS_TEAM_GH}/tf-keras/tree/v2.19.0/",
3939
"keras_rs": f"{KERAS_TEAM_GH}/keras-rs/tree/v0.2.1/"
4040
}

scripts/hub_master.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -697,6 +697,29 @@
697697
},
698698
],
699699
},
700+
{
701+
"path": "dinov2/",
702+
"title": "DINOV2",
703+
"toc": True,
704+
"children": [
705+
{
706+
"path": "dinov2_image_converter",
707+
"title": "DINOV2ImageConverter",
708+
"generate": [
709+
"keras_hub.layers.DINOV2ImageConverter",
710+
"keras_hub.layers.DINOV2ImageConverter.from_preset",
711+
],
712+
},
713+
{
714+
"path": "dinov2_backbone",
715+
"title": "DINOV2Backbone model",
716+
"generate": [
717+
"keras_hub.models.DINOV2Backbone",
718+
"keras_hub.models.DINOV2Backbone.from_preset",
719+
],
720+
},
721+
],
722+
},
700723
{
701724
"path": "distil_bert/",
702725
"title": "DistilBert",

0 commit comments

Comments
 (0)