Added additional information on addine new models and some additional gotchas

This commit is contained in:
Jaret Burkett
2026-06-18 15:04:50 -06:00
parent 515b0ea5cd
commit e886745051
3 changed files with 64 additions and 0 deletions

View File

@@ -56,6 +56,11 @@ class ExampleModel(BaseModel):
# - it is the default cache key for text-embedding / latent caches
arch = "example"
# ALL NEW MODELS should set this to False. ``BaseModel`` defaults it to True
# only for backwards-compatibility with already-released LoKr checkpoints; the
# newer LoKr weight format is the correct one for any new architecture.
use_old_lokr_format = False
def __init__(
self,
device, # "cuda:0" etc.