fix ckpt_path in e4t tuning

brianfitzgerald · brianfitzgerald · commit 841709e0af5b · 2023-04-16T18:00:23.000-05:00
diff --git a/tuning_e4t.py b/tuning_e4t.py
@@ -103,7 +103,7 @@ def main():
     unet = load_e4t_unet(
         pretrained_model_name_or_path=pretrained_args.pretrained_model_name_or_path,
         # load weight offsets from pre-trained model
-        ckpt_path=os.path.join(args.pretrained_model_name_or_path, "weight_offsets.pt"),
+        ckpt_path=args.pretrained_model_name_or_path,
     )
     # encoder
     e4t_encoder = load_e4t_encoder(

Original file line number	Diff line number	Diff line change
`@@ -103,7 +103,7 @@ def main():`
`103`	`103`	`unet = load_e4t_unet(`
`104`	`104`	`pretrained_model_name_or_path=pretrained_args.pretrained_model_name_or_path,`
`105`	`105`	`# load weight offsets from pre-trained model`
`106`		`- ckpt_path=os.path.join(args.pretrained_model_name_or_path, "weight_offsets.pt"),`
	`106`	`+ ckpt_path=args.pretrained_model_name_or_path,`
`107`	`107`	`)`
`108`	`108`	`# encoder`
`109`	`109`	`e4t_encoder = load_e4t_encoder(`