missing dependencies

#27
by bikekowal - opened

Hi all,

I am missing some of the dependencies to be imported. One of them is "normalize" function which I believe should be imported together with torch.

the error is following:

image = normalize(image,[0.5,0.5,0.5],[1.0,1.0,1.0])
NameError: name 'normalize' is not defined

I appreciate your help.
best,
Marcin

BRIA AI org

right, missing the line -
from torchvision.transforms.functional import normalize
i fixed the model card.

Sign up or log in to comment