Upload mgie_llava.py with huggingface_hub

mgie_llava.py

@@ -241,9 +241,9 @@ class LlavaLlamaForCausalLM_(LlamaForCausalLM):
 
         self.edit_head = EditMapper()
 
-        self.scheduler, self.vae, self.unet = [diffusers.DDPMScheduler.from_pretrained('/root/autodl-tmp/_ckpt/stable_diffusion', subfolder='scheduler'),
-                                               diffusers.AutoencoderKL.from_pretrained('/root/autodl-tmp/_ckpt/stable_diffusion', subfolder='vae'),
-                                               diffusers.UNet2DConditionModel.from_pretrained('/root/autodl-tmp/_ckpt/stable_diffusion', subfolder='unet')]
+        self.scheduler, self.vae, self.unet = [diffusers.DDPMScheduler.from_pretrained('/home/zbz5349/WorkSpace/aigeeks/ml-mgie/_ckpt/sd', subfolder='scheduler'),
+                                               diffusers.AutoencoderKL.from_pretrained('/home/zbz5349/WorkSpace/aigeeks/ml-mgie/_ckpt/sd', subfolder='vae'),
+                                               diffusers.UNet2DConditionModel.from_pretrained('/home/zbz5349/WorkSpace/aigeeks/ml-mgie/_ckpt/sd', subfolder='unet')]
         self.vae.requires_grad_(False)
         self.unet.register_to_config(in_channels=8)
         with torch.no_grad():