Spaces:

sdart
/

SD_txt2img

Runtime error

sdart commited on May 16, 2023

Commit

3206fc4

1 Parent(s): 5126a09

switch to gpu

Files changed (1) hide show

model.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 import PIL.Image
 import numpy as np
-device = "cpu"
 class Model:
     def __init__(self):

 import PIL.Image
 import numpy as np
+device = "cuda"
 class Model:
     def __init__(self):