39 lines
1.4 KiB
Plaintext
39 lines
1.4 KiB
Plaintext
stable-diffusion python module
|
|
|
|
docs - http://10.6.9.69:7860/docs
|
|
|
|
testing api - http://10.6.9.69:7860/sdapi/v1
|
|
|
|
Needs to implement a class called DiffuseAPI
|
|
|
|
The class needs be initialised with the following variables
|
|
- url - String
|
|
- nsfw_enabled - Boolean
|
|
- styles - List<Strings>
|
|
- steps - Integer
|
|
|
|
The class will need the following functions
|
|
- generate_image
|
|
- takes in a prompt and a negative_prompt (negative prompt will default to "")
|
|
- returns the base64 of the returned image or returns the error
|
|
|
|
- set_seed
|
|
- takes in an integer and updates the seed in the payload with the provided seed
|
|
- set_steps
|
|
- takes in an integer and updates the steps in the payload with the provided steps
|
|
- set_cfg_scale
|
|
- takes in an integer and updates the cfg_scale in the payload with the provided cfg_scale
|
|
- set_styles
|
|
- takes in a list of strings and updates the styles in the payload with the provided styles
|
|
- set_orientation
|
|
- take in one of three options [portrait, landscape, square] and set the width and height in the payload accordingly
|
|
- get_orientation
|
|
- return the current orientation and heigh + width
|
|
- set_nsfw_filter
|
|
- takes in a bool and updates the nsfw filter accordingly
|
|
- get_nsfw_filter
|
|
- returns the current state of the nsfw filter
|
|
|
|
- upscale_image
|
|
- takes in the image (in b64) and sends it to the upscale endpoint
|
|
- returns the upscaled image |