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
105 stars 7 forks source link

Make it possible to save SPAN models as `.safetensors` #157

Closed RunDevelopment closed 4 months ago

RunDevelopment commented 4 months ago

Calling .contiguous() for the tensors we assign as weight and bias makes safetensors happy.