刘虹雨 commited on
Commit
d2f1ed6
·
1 Parent(s): 648d75e

update code

Browse files
.DS_Store CHANGED
Binary files a/.DS_Store and b/.DS_Store differ
 
DiT_VAE/.DS_Store CHANGED
Binary files a/DiT_VAE/.DS_Store and b/DiT_VAE/.DS_Store differ
 
DiT_VAE/diffusion/.DS_Store CHANGED
Binary files a/DiT_VAE/diffusion/.DS_Store and b/DiT_VAE/diffusion/.DS_Store differ
 
DiT_VAE/diffusion/data/.DS_Store CHANGED
Binary files a/DiT_VAE/diffusion/data/.DS_Store and b/DiT_VAE/diffusion/data/.DS_Store differ
 
app.py CHANGED
@@ -1048,14 +1048,8 @@ if __name__ == '__main__':
1048
  controlnet_path, torch_dtype=torch.float16
1049
  )
1050
  sd_path = './pretrained_model/sd21'
1051
- text_encoder = transformers.CLIPTextModel.from_pretrained(
1052
- sd_path,
1053
- subfolder="text_encoder",
1054
- num_hidden_layers=12 - (2 - 1),
1055
- torch_dtype=torch.float16
1056
- )
1057
  pipeline_sd = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
1058
- sd_path, torch_dtype=torch.float16, text_encoder=text_encoder,
1059
  use_safetensors=True, controlnet=controlnet, variant="fp16"
1060
  ).to(device)
1061
  pipeline_sd.scheduler=DPMSolverMultistepScheduler.from_config(pipeline_sd.scheduler.config, use_karras_sigmas=True)
 
1048
  controlnet_path, torch_dtype=torch.float16
1049
  )
1050
  sd_path = './pretrained_model/sd21'
 
 
 
 
 
 
1051
  pipeline_sd = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
1052
+ sd_path, torch_dtype=torch.float16,
1053
  use_safetensors=True, controlnet=controlnet, variant="fp16"
1054
  ).to(device)
1055
  pipeline_sd.scheduler=DPMSolverMultistepScheduler.from_config(pipeline_sd.scheduler.config, use_karras_sigmas=True)
data_process/.DS_Store CHANGED
Binary files a/data_process/.DS_Store and b/data_process/.DS_Store differ
 
demo_data/.DS_Store CHANGED
Binary files a/demo_data/.DS_Store and b/demo_data/.DS_Store differ
 
demo_data/target_video/.DS_Store CHANGED
Binary files a/demo_data/target_video/.DS_Store and b/demo_data/target_video/.DS_Store differ
 
demo_data/target_video/data_Obama/.DS_Store CHANGED
Binary files a/demo_data/target_video/data_Obama/.DS_Store and b/demo_data/target_video/data_Obama/.DS_Store differ
 
demo_data/target_video/data_Obama/coeffs/.DS_Store CHANGED
Binary files a/demo_data/target_video/data_Obama/coeffs/.DS_Store and b/demo_data/target_video/data_Obama/coeffs/.DS_Store differ
 
demo_data/target_video/data_Obama/images512x512/.DS_Store ADDED
Binary file (6.15 kB). View file
 
demo_data/target_video/data_Obama/motions/.DS_Store CHANGED
Binary files a/demo_data/target_video/data_Obama/motions/.DS_Store and b/demo_data/target_video/data_Obama/motions/.DS_Store differ
 
recon/.DS_Store CHANGED
Binary files a/recon/.DS_Store and b/recon/.DS_Store differ