Biswanath Saha

@patricklabatut thsnks for the reply. I managed to fix it or rather I should say, it kind of fixed itself. I checked the CUDA version is installed correctly. The xformer...

Try this, pip install --pre -U xformers. The dev build of xformers have 0.0.18

How about this?? > img = Image.open('') > transform = transforms.Compose([ > transforms.Resize(256), > transforms.CenterCrop(224), > transforms.ToTensor(), > transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]), ]) > input_tensor = transform(img) >...

Thank you for getting back to me. Will you be releasing any training script, I would love to fine-tune the model for my images