From 69a3b22fa1b48aab9cb41e85aa3d3c98ca8a7fa9 Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Fri, 30 Aug 2024 10:04:36 +0200 Subject: [PATCH] chore(tests): replace runaway models for tests (#3432) Signed-off-by: Ettore Di Giacinto --- backend/python/diffusers/test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/python/diffusers/test.py b/backend/python/diffusers/test.py index b5e381ba..0e92538e 100644 --- a/backend/python/diffusers/test.py +++ b/backend/python/diffusers/test.py @@ -53,7 +53,7 @@ class TestBackendServicer(unittest.TestCase): self.setUp() with grpc.insecure_channel("localhost:50051") as channel: stub = backend_pb2_grpc.BackendStub(channel) - response = stub.LoadModel(backend_pb2.ModelOptions(Model="runwayml/stable-diffusion-v1-5")) + response = stub.LoadModel(backend_pb2.ModelOptions(Model="Lykon/dreamshaper-8")) self.assertTrue(response.success) self.assertEqual(response.message, "Model loaded successfully") except Exception as err: @@ -71,7 +71,7 @@ class TestBackendServicer(unittest.TestCase): self.setUp() with grpc.insecure_channel("localhost:50051") as channel: stub = backend_pb2_grpc.BackendStub(channel) - response = stub.LoadModel(backend_pb2.ModelOptions(Model="runwayml/stable-diffusion-v1-5")) + response = stub.LoadModel(backend_pb2.ModelOptions(Model="Lykon/dreamshaper-8")) print(response.message) self.assertTrue(response.success) image_req = backend_pb2.GenerateImageRequest(positive_prompt="cat", width=16,height=16, dst="test.jpg") @@ -81,4 +81,4 @@ class TestBackendServicer(unittest.TestCase): print(err) self.fail("Image gen service failed") finally: - self.tearDown() \ No newline at end of file + self.tearDown()