holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added a clear error message when targeting CPU if not enabled in config.

Browse Source
This commit is contained in:
Marc-Andre Ferland 2022-10-16 22:24:26 -04:00
parent 7c72608e1c
commit 994d62ac65
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ui/server.py
View File

@ -244,6 +244,7 @@ def save_model_to_config(model_name):
@app.post('/render') @app.post('/render')
def render(req : task_manager.ImageRequest): def render(req : task_manager.ImageRequest):
if req.use_cpu and task_manager.is_alive('cpu') <= 0: return HTTPException(status_code=403, detail=f'CPU rendering is not enabled in config.json or the thread has died...') # HTTP403 Forbidden
try: try:
save_model_to_config(req.use_stable_diffusion_model) save_model_to_config(req.use_stable_diffusion_model)
req.use_stable_diffusion_model = resolve_model_to_use(req.use_stable_diffusion_model) req.use_stable_diffusion_model = resolve_model_to_use(req.use_stable_diffusion_model)