Yaron Koresh commited on
Commit
a0f86a3
·
verified ·
1 Parent(s): b8d8aa1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -51,8 +51,8 @@ dtype = torch.float16
51
  device = "cuda"
52
  #repo = "ByteDance/AnimateDiff-Lightning"
53
  #ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
54
- base = "emilianJR/epiCRealism"
55
- #base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
56
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(device, dtype=dtype)
57
  #unet = UNet2DConditionModel.from_config("emilianJR/epiCRealism",subfolder="unet").to(device, dtype).load_state_dict(load_file(hf_hub_download("emilianJR/epiCRealism", "unet/diffusion_pytorch_model.safetensors"), device=device), strict=False)
58
  adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
@@ -67,8 +67,8 @@ result = []
67
  fast=True
68
  fps=15
69
  time=3
70
- width=896
71
- height=896
72
  step=40
73
  accu=10
74
 
 
51
  device = "cuda"
52
  #repo = "ByteDance/AnimateDiff-Lightning"
53
  #ckpt = f"animatediff_lightning_{step}step_diffusers.safetensors"
54
+ #base = "emilianJR/epiCRealism"
55
+ base = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
56
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to(device, dtype=dtype)
57
  #unet = UNet2DConditionModel.from_config("emilianJR/epiCRealism",subfolder="unet").to(device, dtype).load_state_dict(load_file(hf_hub_download("emilianJR/epiCRealism", "unet/diffusion_pytorch_model.safetensors"), device=device), strict=False)
58
  adapter = MotionAdapter.from_pretrained("guoyww/animatediff-motion-adapter-v1-5-3", torch_dtype=dtype, device=device)
 
67
  fast=True
68
  fps=15
69
  time=3
70
+ width=768
71
+ height=1024
72
  step=40
73
  accu=10
74