search

chaiNNer-org / spandrel

Spandrel gives your project support for various PyTorch architectures meant for AI Super-Resolution, restoration, and inpainting. Based on the model support implemented in chaiNNer.
MIT License
112 stars 7 forks source link

use the model seems incorrect #103

Closed whybfq closed 6 months ago

whybfq commented 6 months ago

use the model

def process(image: torch.Tensor) -> torch.Tensor: with torch.no_grad(): return model(tensor)

should the last line change to return model(image)

joeyballentine commented 6 months ago

Yes that's a typo, nice catch