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 = [
|