finegrain-ai / refiners

A microframework on top of PyTorch with first-class citizen APIs for foundation model adaptation
https://refine.rs
MIT License
585 stars 43 forks source link

MultiUpscaler: specify map_location when loading negative embedding #387

Closed Laurent2916 closed 2 months ago

Laurent2916 commented 2 months ago

Forgot the map_location=self.device