JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-04-06 03:57:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

padding fix

Browse Source
This commit is contained in:
u 2024-09-08 17:09:21 +08:00
parent 98466e674c
commit 202484e77f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference/gradio_composite_demo/rife_model.py
View File

@ -19,7 +19,7 @@ def pad_image(img, scale):
tmp = max(32, int(32 / scale)) tmp = max(32, int(32 / scale))
ph = ((h - 1) // tmp + 1) * tmp ph = ((h - 1) // tmp + 1) * tmp
pw = ((w - 1) // tmp + 1) * tmp pw = ((w - 1) // tmp + 1) * tmp
padding = (0, 0, pw - w, ph - h) padding = (0, pw - w, 0, ph - h)
return F.pad(img, padding) return F.pad(img, padding)