diff --git a/gligen/trainer.py b/gligen/trainer.py index 0f82218f..9f295e53 100644 --- a/gligen/trainer.py +++ b/gligen/trainer.py @@ -60,7 +60,7 @@ def __call__(self, images, real, masked_real, captions, seen): def read_official_ckpt(ckpt_path): - "Read offical pretrained SD ckpt and convert into my style" + "Read official pretrained SD ckpt and convert into my style" state_dict = torch.load(ckpt_path, map_location="cpu")["state_dict"] out = {} out["model"] = {}