Gflorent
/

blip_captioning

image-captioning

endpoints-template

Model card Files Files and versions

Gflorent commited on Aug 16, 2022

Commit

8f294b5

·

1 Parent(s): 3dedfc7

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -9,7 +9,7 @@ from torchvision import transforms
 from torchvision.transforms.functional import InterpolationMode
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 class PreTrainedPipeline():
     def __init__(self, path=""):

 from torchvision.transforms.functional import InterpolationMode
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+print(device)
 class PreTrainedPipeline():
     def __init__(self, path=""):