Fix legacy model loading

This commit is contained in:
somebody
2023-04-15 12:57:35 -05:00
parent 2a977feb3e
commit 4dcf570407
4 changed files with 36 additions and 16 deletions

View File

@@ -17,6 +17,9 @@ from modeling import logits_processors
import utils
class SuperLegacyModelError(RuntimeError):
pass
# We only want to use logit manipulations and such on our core text model
class use_core_manipulations:
"""Use in a `with` block to patch functions for core story model sampling."""