File tree Expand file tree Collapse file tree 15 files changed +1
-2408
lines changed
.kokoro/github/ubuntu/gpu
tools/checkpoint_conversion Expand file tree Collapse file tree 15 files changed +1
-2408
lines changed Original file line number Diff line number Diff line change 38
38
pip install torch>=2.0.1+cpu
39
39
pip install "jax[cpu]"
40
40
pip install keras-core
41
- pip install keras-nlp-nightly --no-deps
42
- pip install tensorflow-text==2.15
43
41
pip install -e ".[tests]" --progress-bar off --upgrade
44
42
- name : Test with pytest
45
43
env :
77
75
run : |
78
76
pip install -r requirements.txt
79
77
pip install -e ".[tests]" --progress-bar off --upgrade
80
- pip install keras-nlp-nightly
81
78
- name : Test with pytest
82
79
env :
83
80
TEST_CUSTOM_OPS : false # TODO(ianstenbit): test custom ops, or figure out what our story is here
Original file line number Diff line number Diff line change 29
29
pip install --extra-index-url https://download.pytorch.org/whl/cpu torch==2.1.0+cpu
30
30
pip install torchvision~=0.16.0
31
31
pip install " jax[cpu]"
32
- pip install keras-nlp-nightly --no-deps
33
- pip install tensorflow-text==2.15
34
32
35
33
elif [ " $KERAS_BACKEND " == " tensorflow" ]
36
34
then
37
35
echo " TensorFlow backend detected."
38
36
pip install -r requirements-tensorflow-cuda.txt --progress-bar off
39
- pip install keras-nlp-nightly
40
37
41
38
elif [ " $KERAS_BACKEND " == " jax" ]
42
39
then
43
40
echo " JAX backend detected."
44
41
pip install -r requirements-jax-cuda.txt --progress-bar off
45
- pip install keras-nlp-nightly
46
42
47
43
elif [ " $KERAS_BACKEND " == " torch" ]
48
44
then
49
45
echo " PyTorch backend detected."
50
46
pip install -r requirements-torch-cuda.txt --progress-bar off
51
- pip install keras-nlp-nightly
52
47
fi
53
48
54
49
pip install --no-deps -e " ." --progress-bar off
Original file line number Diff line number Diff line change 183
183
from keras_cv .models .backbones .vit_det .vit_det_aliases import ViTDetLBackbone
184
184
from keras_cv .models .backbones .vit_det .vit_det_backbone import ViTDetBackbone
185
185
from keras_cv .models .classification .image_classifier import ImageClassifier
186
- from keras_cv .models .feature_extractor .clip import CLIP
187
186
from keras_cv .models .object_detection .retinanet .retinanet import RetinaNet
188
187
from keras_cv .models .object_detection .yolo_v8 .yolo_v8_backbone import (
189
188
YOLOV8Backbone ,
Load Diff This file was deleted.
Load Diff This file was deleted.
You can’t perform that action at this time.
0 commit comments