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 3838 pip install torch>=2.0.1+cpu
3939 pip install "jax[cpu]"
4040 pip install keras-core
41- pip install keras-nlp-nightly --no-deps
42- pip install tensorflow-text==2.15
4341 pip install -e ".[tests]" --progress-bar off --upgrade
4442 - name : Test with pytest
4543 env :
7775 run : |
7876 pip install -r requirements.txt
7977 pip install -e ".[tests]" --progress-bar off --upgrade
80- pip install keras-nlp-nightly
8178 - name : Test with pytest
8279 env :
8380 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 2929 pip install --extra-index-url https://download.pytorch.org/whl/cpu torch==2.1.0+cpu
3030 pip install torchvision~=0.16.0
3131 pip install " jax[cpu]"
32- pip install keras-nlp-nightly --no-deps
33- pip install tensorflow-text==2.15
3432
3533elif [ " $KERAS_BACKEND " == " tensorflow" ]
3634then
3735 echo " TensorFlow backend detected."
3836 pip install -r requirements-tensorflow-cuda.txt --progress-bar off
39- pip install keras-nlp-nightly
4037
4138elif [ " $KERAS_BACKEND " == " jax" ]
4239then
4340 echo " JAX backend detected."
4441 pip install -r requirements-jax-cuda.txt --progress-bar off
45- pip install keras-nlp-nightly
4642
4743elif [ " $KERAS_BACKEND " == " torch" ]
4844then
4945 echo " PyTorch backend detected."
5046 pip install -r requirements-torch-cuda.txt --progress-bar off
51- pip install keras-nlp-nightly
5247fi
5348
5449pip install --no-deps -e " ." --progress-bar off
Original file line number Diff line number Diff line change 183183from keras_cv .models .backbones .vit_det .vit_det_aliases import ViTDetLBackbone
184184from keras_cv .models .backbones .vit_det .vit_det_backbone import ViTDetBackbone
185185from keras_cv .models .classification .image_classifier import ImageClassifier
186- from keras_cv .models .feature_extractor .clip import CLIP
187186from keras_cv .models .object_detection .retinanet .retinanet import RetinaNet
188187from keras_cv .models .object_detection .yolo_v8 .yolo_v8_backbone import (
189188 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