feat(diffusers): various enhancements (#895)

mudler · Aug 14, 2023 · a96c3bc · a96c3bc
1 parent 77e1ae3
commit a96c3bc
Show file tree

Hide file tree

Showing 11 changed files with 165 additions and 101 deletions.
diff --git a/api/backend/image.go b/api/backend/image.go
@@ -39,14 +39,15 @@ func ImageGeneration(height, width, mode, step, seed int, positive_prompt, negat
 		_, err := inferenceModel.GenerateImage(
 			o.Context,
 			&proto.GenerateImageRequest{
-				Height:         int32(height),
-				Width:          int32(width),
-				Mode:           int32(mode),
-				Step:           int32(step),
-				Seed:           int32(seed),
-				PositivePrompt: positive_prompt,
-				NegativePrompt: negative_prompt,
-				Dst:            dst,
+				Height:           int32(height),
+				Width:            int32(width),
+				Mode:             int32(mode),
+				Step:             int32(step),
+				Seed:             int32(seed),
+				PositivePrompt:   positive_prompt,
+				NegativePrompt:   negative_prompt,
+				Dst:              dst,
+				EnableParameters: c.Diffusers.EnableParameters,
 			})
 		return err
 	}

diff --git a/api/config/config.go b/api/config/config.go
@@ -42,9 +42,10 @@ type Config struct {
 }
 
 type Diffusers struct {
-	PipelineType  string `yaml:"pipeline_type"`
-	SchedulerType string `yaml:"scheduler_type"`
-	CUDA          bool   `yaml:"cuda"`
+	PipelineType     string `yaml:"pipeline_type"`
+	SchedulerType    string `yaml:"scheduler_type"`
+	CUDA             bool   `yaml:"cuda"`
+	EnableParameters string `yaml:"enable_parameters"` // A list of comma separated parameters to specify
 }
 
 type LLMConfig struct {

diff --git a/extra/grpc/autogptq/backend_pb2.py b/extra/grpc/autogptq/backend_pb2.py