apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix conditional tests

This commit is contained in:
Sean Sube 2023-09-25 22:10:47 -05:00
parent 23aa00d696
commit 898d76e4a5
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/tests/test_diffusers/test_load.py
View File

@ -172,7 +172,7 @@ class TestLoadControlNet(unittest.TestCase):
class TestLoadTextEncoders(unittest.TestCase): class TestLoadTextEncoders(unittest.TestCase):
@unittest.skipUnless(lambda: path.exists("../models/stable-diffusion-onnx-v1-5/text_encoder/model.onnx"), "model does not exist") @unittest.skipUnless(path.exists("../models/stable-diffusion-onnx-v1-5/text_encoder/model.onnx"), "model does not exist")
def test_load_embeddings(self): def test_load_embeddings(self):
""" """
Should add the token to tokenizer Should add the token to tokenizer
@ -194,7 +194,7 @@ class TestLoadTextEncoders(unittest.TestCase):
def test_load_embeddings_xl(self): def test_load_embeddings_xl(self):
pass pass
@unittest.skipUnless(lambda: path.exists("../models/stable-diffusion-onnx-v1-5/text_encoder/model.onnx"), "model does not exist") @unittest.skipUnless(path.exists("../models/stable-diffusion-onnx-v1-5/text_encoder/model.onnx"), "model does not exist")
def test_load_loras(self): def test_load_loras(self):
components = load_text_encoders( components = load_text_encoders(
ServerContext(model_path="../models"), ServerContext(model_path="../models"),
@ -213,7 +213,7 @@ class TestLoadTextEncoders(unittest.TestCase):
pass pass
class TestLoadUnet(unittest.TestCase): class TestLoadUnet(unittest.TestCase):
@unittest.skipUnless(lambda: path.exists("../models/stable-diffusion-onnx-v1-5/unet/model.onnx"), "model does not exist") @unittest.skipUnless(path.exists("../models/stable-diffusion-onnx-v1-5/unet/model.onnx"), "model does not exist")
def test_load_unet_loras(self): def test_load_unet_loras(self):
components = load_unet( components = load_unet(
ServerContext(model_path="../models"), ServerContext(model_path="../models"),
@ -230,7 +230,7 @@ class TestLoadUnet(unittest.TestCase):
def test_load_unet_loras_xl(self): def test_load_unet_loras_xl(self):
pass pass
@unittest.skipUnless(lambda: path.exists("../models/stable-diffusion-onnx-v1-5/cnet/model.onnx"), "model does not exist") @unittest.skipUnless(path.exists("../models/stable-diffusion-onnx-v1-5/cnet/model.onnx"), "model does not exist")
def test_load_cnet_loras(self): def test_load_cnet_loras(self):
components = load_unet( components = load_unet(
ServerContext(model_path="../models"), ServerContext(model_path="../models"),
@ -246,7 +246,7 @@ class TestLoadUnet(unittest.TestCase):
class TestLoadVae(unittest.TestCase): class TestLoadVae(unittest.TestCase):
@unittest.skipUnless(lambda: path.exists("../models/upscaling-stable-diffusion-x4/vae/model.onnx"), "model does not exist") @unittest.skipUnless(path.exists("../models/upscaling-stable-diffusion-x4/vae/model.onnx"), "model does not exist")
def test_load_single(self): def test_load_single(self):
""" """
Should return single component Should return single component
@ -261,7 +261,7 @@ class TestLoadVae(unittest.TestCase):
self.assertNotIn("vae_decoder", components) self.assertNotIn("vae_decoder", components)
self.assertNotIn("vae_encoder", components) self.assertNotIn("vae_encoder", components)
@unittest.skipUnless(lambda: path.exists("../models/stable-diffusion-onnx-v1-5/vae_encoder/model.onnx"), "model does not exist") @unittest.skipUnless(path.exists("../models/stable-diffusion-onnx-v1-5/vae_encoder/model.onnx"), "model does not exist")
def test_load_split(self): def test_load_split(self):
""" """
Should return split encoder/decoder Should return split encoder/decoder