Skip to content

Allow for using CPU if no CUDA device is detected #56

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ModeratePrawn wants to merge 7 commits into
base: main
Choose a base branch
from
Open

Allow for using CPU if no CUDA device is detected #56

Changes from 1 commit
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
feae820
Add files via upload
ModeratePrawn Aug 22, 2022
333e21e
Merge pull request #1 from ModeratePrawn/ModeratePrawn-patch-1
ModeratePrawn Aug 22, 2022
d3f1ebc
added cpu support to img2img.py
SpandexWizard Sep 1, 2022
88f919c
Add files via upload
ModeratePrawn Sep 1, 2022
d3e6e36
Delete img2img.py
ModeratePrawn Sep 1, 2022
d68cd0d
Add files via upload
ModeratePrawn Sep 1, 2022
a32c453
Merge pull request #2 from SpandexWizard/main
ModeratePrawn Sep 1, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
added cpu support to img2img.py
  • Loading branch information
@SpandexWizard
SpandexWizard authored Sep 1, 2022
commit d3f1ebcbd71ec5521142c55f617e92303ffd872d
3 changes: 2 additions & 1 deletion scripts/img2img.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,8 @@ def load_model_from_config(config, ckpt, verbose=False):
print("unexpected keys:")
print(u)

model.cuda()
if torch.cuda.is_available():
model.cuda()
model.eval()
return model

Expand Down