diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-04 12:20:10 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-04 12:20:10 +0000 |
| commit | bc4d457de82f76f8ab9f2bedf933c06deb5d5ba9 (patch) | |
| tree | 969e0e595bd36987ae9de9ae302085ef555bba15 /modules/gfpgan_model.py | |
| parent | d5bba20a58f43a9f984bb67b4e17f48661f6b818 (diff) | |
| parent | e9e2a7ec9ac704f133f586eb34176e388c93c87c (diff) | |
| download | stable-diffusion-webui-gfx803-bc4d457de82f76f8ab9f2bedf933c06deb5d5ba9.tar.gz stable-diffusion-webui-gfx803-bc4d457de82f76f8ab9f2bedf933c06deb5d5ba9.tar.bz2 stable-diffusion-webui-gfx803-bc4d457de82f76f8ab9f2bedf933c06deb5d5ba9.zip | |
Merge pull request #1616 from brkirch/cpu-cmdline-opt
Add --use-cpu command line option
Diffstat (limited to 'modules/gfpgan_model.py')
| -rw-r--r-- | modules/gfpgan_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/gfpgan_model.py b/modules/gfpgan_model.py index 5586b554..a9452dce 100644 --- a/modules/gfpgan_model.py +++ b/modules/gfpgan_model.py @@ -21,7 +21,7 @@ def gfpgann(): global loaded_gfpgan_model
global model_path
if loaded_gfpgan_model is not None:
- loaded_gfpgan_model.gfpgan.to(shared.device)
+ loaded_gfpgan_model.gfpgan.to(devices.device_gfpgan)
return loaded_gfpgan_model
if gfpgan_constructor is None:
@@ -53,7 +53,7 @@ def gfpgan_fix_faces(np_image): if model is None:
return np_image
- send_model_to(model, devices.device)
+ send_model_to(model, devices.device_gfpgan)
np_image_bgr = np_image[:, :, ::-1]
cropped_faces, restored_faces, gfpgan_output_bgr = model.enhance(np_image_bgr, has_aligned=False, only_center_face=False, paste_back=True)
|