4 Commits

Author	SHA1	Message	Date
psychedelicious	99a8ebe3a0	chore: ruff check - fix flake8-bugbear	2023-11-11 10:55:28 +11:00
Ryan Dick	e391f3c9a8	Skip torch.nn.Embedding.reset_parameters(...) when loading a text encoder model.	2023-11-02 19:41:33 -07:00
Ryan Dick	61242bf86a	Fix bug in skip_torch_weight_init() where the original behavior of torch.nn.Conv*d modules wasn't being restored correctly.	2023-10-10 10:05:50 -04:00
Ryan Dick	58b56e9b1e	Add a skip_torch_weight_init() context manager to improve model load times (from disk).	2023-10-09 14:12:56 -04:00