fix for 'model is not defined' when loading embedding

This commit is contained in:
Jan Skurovec 2022-10-19 08:54:47 +02:00 committed by Lincoln Stein
parent 213e12fe13
commit d4d1014c9f

View File

@ -728,7 +728,7 @@ class Generate:
seed_everything(random.randrange(0, np.iinfo(np.uint32).max))
if self.embedding_path is not None:
model.embedding_manager.load(
self.model.embedding_manager.load(
self.embedding_path, self.precision == 'float32' or self.precision == 'autocast'
)