Spaces:
Sleeping
Sleeping
Reverted app to cpu
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ build_custom_prompter()
|
|
43 |
from gradio_image_prompter import ImagePrompter
|
44 |
#print("torch version")
|
45 |
#print(torch.version.cuda)
|
46 |
-
install_add_dependencies()
|
47 |
|
48 |
class AppSteps(Enum):
|
49 |
JUST_TEXT = 1
|
@@ -166,7 +166,7 @@ args = parser.parse_args()
|
|
166 |
# args.device = torch.device('cuda')
|
167 |
#else:
|
168 |
# args.device = torch.device('cpu')
|
169 |
-
args.device = torch.device('
|
170 |
model, transform = build_model_and_transforms(args)
|
171 |
|
172 |
examples = [
|
|
|
43 |
from gradio_image_prompter import ImagePrompter
|
44 |
#print("torch version")
|
45 |
#print(torch.version.cuda)
|
46 |
+
#install_add_dependencies()
|
47 |
|
48 |
class AppSteps(Enum):
|
49 |
JUST_TEXT = 1
|
|
|
166 |
# args.device = torch.device('cuda')
|
167 |
#else:
|
168 |
# args.device = torch.device('cpu')
|
169 |
+
args.device = torch.device('cpu')
|
170 |
model, transform = build_model_and_transforms(args)
|
171 |
|
172 |
examples = [
|