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

This commit is contained in:
Marc-Andre Ferland 2022-10-16 22:24:26 -04:00
parent 7c72608e1c
commit 994d62ac65

View File

@ -244,6 +244,7 @@ def save_model_to_config(model_name):
@app.post('/render')
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:
save_model_to_config(req.use_stable_diffusion_model)
req.use_stable_diffusion_model = resolve_model_to_use(req.use_stable_diffusion_model)