tests with AutoPipeline

Author: mudler

backend/python/diffusers/backend_diffusers.py

@@ -136,7 +136,7 @@ def LoadModel(self, request, context):
             print(f"Loading model {request.Model}...", file=sys.stderr)
             print(f"Request {request}", file=sys.stderr)
             torchType = torch.float32
-            variant = ""
+            variant = None
 
             if request.F16Memory:
                 torchType = torch.float16
@@ -165,8 +165,9 @@ def LoadModel(self, request, context):
             fromSingleFile = request.Model.startswith("http") or request.Model.startswith("/") or local
 
 
+            # If PipelineType is not defined, defaults to AutoPipelineForText2Image
             if request.PipelineType == "":
-                request.PipelineType == "AutoPipelineForText2Image"
+                request.PipelineType = "AutoPipelineForText2Image"
 
             ## img2img
             if (request.PipelineType == "StableDiffusionImg2ImgPipeline") or (request.IMG2IMG and request.PipelineType == ""):

backend/python/diffusers/test.py

@@ -53,7 +53,7 @@ def test_load_model(self):
             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", PipelineType="StableDiffusionPipeline"))
+                response = stub.LoadModel(backend_pb2.ModelOptions(Model="runwayml/stable-diffusion-v1-5"))
                 self.assertTrue(response.success)
                 self.assertEqual(response.message, "Model loaded successfully")
         except Exception as err:
@@ -71,7 +71,7 @@ def test(self):
             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", PipelineType="StableDiffusionPipeline"))
+                response = stub.LoadModel(backend_pb2.ModelOptions(Model="runwayml/stable-diffusion-v1-5"))
                 print(response.message)
                 self.assertTrue(response.success)
                 image_req = backend_pb2.GenerateImageRequest(positive_prompt="cat", width=16,height=16, dst="test.jpg")