From cb58cfa64fe36c84ebf70620812be9638ef7f3da Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Tue, 7 Jun 2022 21:47:23 -0700 Subject: [PATCH] Skip pretrained models for <1.7 --- lib/Boltz/test/vision.jl | 1 + 1 file changed, 1 insertion(+) diff --git a/lib/Boltz/test/vision.jl b/lib/Boltz/test/vision.jl index 1860a1996..1c2402687 100644 --- a/lib/Boltz/test/vision.jl +++ b/lib/Boltz/test/vision.jl @@ -65,6 +65,7 @@ models_available = Dict(alexnet => [(:alexnet, true), (:alexnet, false)], @time begin @testset "name = $name & pretrained = $pretrained" for (name, pretrained) in config if VERSION <= v"1.7" && pretrained @warn "Skipping pretrained models in Julia < 1.7" + continue end model, ps, st = model_creator(name; pretrained) st = Lux.testmode(st)