Fix push_to_hub for dreambooth and textual_inversion

Author: YaYaB

examples/dreambooth/train_dreambooth.py

@@ -14,6 +14,7 @@
 from accelerate.utils import set_seed
 from diffusers import AutoencoderKL, DDPMScheduler, StableDiffusionPipeline, UNet2DConditionModel
 from diffusers.optimization import get_scheduler
+from diffusers.hub_utils import push_to_hub
 from huggingface_hub import HfFolder, Repository, whoami
 from PIL import Image
 from torchvision import transforms
@@ -575,7 +576,7 @@ def collate_fn(examples):
         pipeline.save_pretrained(args.output_dir)
 
         if args.push_to_hub:
-            repo.push_to_hub(
+            push_to_hub(
                 args, pipeline, repo, commit_message="End of training", blocking=False, auto_lfs_prune=True
             )
 

examples/textual_inversion/textual_inversion.py

@@ -19,6 +19,7 @@
 from diffusers import AutoencoderKL, DDPMScheduler, PNDMScheduler, StableDiffusionPipeline, UNet2DConditionModel
 from diffusers.optimization import get_scheduler
 from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
+from diffusers.hub_utils import push_to_hub
 from huggingface_hub import HfFolder, Repository, whoami
 from PIL import Image
 from torchvision import transforms
@@ -569,7 +570,7 @@ def main():
         save_progress(text_encoder, placeholder_token_id, accelerator, args)
 
         if args.push_to_hub:
-            repo.push_to_hub(
+            push_to_hub(
                 args, pipeline, repo, commit_message="End of training", blocking=False, auto_lfs_prune=True
             )