ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-13 04:28:10 +00:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

feat(diffusers): various enhancements (#895)

Browse Source
This commit is contained in:
Ettore Di Giacinto
2023-08-14 23:12:00 +02:00
committed by GitHub
parent 77e1ae3d70
commit a96c3bc885
11 changed files with 165 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
api/backend/image.go
View File

@ -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
}

7
api/config/config.go
View File

@ -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 {