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

Workaround to run gfpgan on cuda:0 even if it's not enabled in the multi-gpu setup

Browse Source
This commit is contained in:
cmdr2
2022-11-14 11:51:18 +05:30
parent 5f880a179c
commit f6651b03b5
3 changed files with 13 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ui/server.py
View File

@ -307,9 +307,6 @@ def update_render_threads_on_request(req : task_manager.ImageRequest):
def render(req : task_manager.ImageRequest):
update_render_threads_on_request(req)
if req.use_face_correction and task_manager.is_alive('cuda:0') <= 0: #TODO Remove when GFPGANer is fixed upstream.
raise HTTPException(status_code=412, detail=f'The "Fix incorrect faces" feature works only on cuda:0. Disable "Fix incorrect faces" (in Image Settings), or use the CUDA_VISIBLE_DEVICES environment variable.')
try:
save_model_to_config(req.use_stable_diffusion_model, req.use_vae_model)
req.use_stable_diffusion_model = resolve_ckpt_to_use(req.use_stable_diffusion_model)