Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
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 |
-
|
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=
|
71 |
-
height=
|
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 |
|