first commit

This commit is contained in:
Benjamyn Love 2022-12-09 12:06:29 +11:00
commit 3eb875d123
2 changed files with 148 additions and 0 deletions

39
api requirements.txt Normal file
View File

@ -0,0 +1,39 @@
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

109
testing.ipynb Normal file

File diff suppressed because one or more lines are too long