Skip to content

Commit 7011960

Browse files
magnusviriJames Reynolds
andauthored
Issue 270 fix (#274)
* check if torch.backends has mps before calling it * Fixes issue 270 Co-authored-by: James Reynolds <[email protected]>
1 parent 0433b3d commit 7011960

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/orig_scripts/img2img.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ def main():
200200
config = OmegaConf.load(f"{opt.config}")
201201
model = load_model_from_config(config, f"{opt.ckpt}")
202202

203-
device = choose_torch_device()
203+
device = torch.device(choose_torch_device())
204204
model = model.to(device)
205205

206206
if opt.plms:

0 commit comments

Comments
 (0)