AUTOMATIC1111/stable-diffusion-webui-promptgen

Promptgen API

dumdiedum opened this issue · 3 comments

Thanks for your work!!

Is it possible to add an api definition, which will extend the current a1111 api?

I took a shot at this with this PR #14

@ngocnv371 i ran into a problem while running the API, can we specify which model to use ?? in API i tried extending the GenerateRequest class but it didn't work, getting error i.e. AttributeError: type object 'GenerateRequest' has no attribute 'model'

image

@mughal41 I'm sorry dude, it was so long ago that my brain wiped all relevant info. But it looks like you're trying to modify the request, you should make sure what you're adding is relevant with the BaseModel. If you want it to load a specific model you need to add that logic handler too, not just squeze it in GeenrateRequest.
Good luck and push a PR once you done?