From dafdd3d9d8611e70176c9454c19d3a82ede3e4ff Mon Sep 17 00:00:00 2001 From: Believethehype Date: Mon, 15 Jan 2024 10:58:16 +0100 Subject: [PATCH] close files before remove --- nostr_dvm/backends/nova_server/utils.py | 2 ++ nostr_dvm/tasks/videogeneration_svd.py | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/nostr_dvm/backends/nova_server/utils.py b/nostr_dvm/backends/nova_server/utils.py index 448cb2a..ce92e1a 100644 --- a/nostr_dvm/backends/nova_server/utils.py +++ b/nostr_dvm/backends/nova_server/utils.py @@ -91,9 +91,11 @@ def check_server_status(jobID, address) -> str | pd.DataFrame: elif content_type == 'video/mp4': with open('./outputs/video.mp4', 'wb') as f: f.write(response.content) + f.close() clip = VideoFileClip("./outputs/video.mp4") clip.write_videofile("./outputs/video2.mp4") result = upload_media_to_hoster("./outputs/video2.mp4") + clip.close() os.remove("./outputs/video.mp4") os.remove("./outputs/video2.mp4") return result diff --git a/nostr_dvm/tasks/videogeneration_svd.py b/nostr_dvm/tasks/videogeneration_svd.py index d7dd732..cbaadfe 100644 --- a/nostr_dvm/tasks/videogeneration_svd.py +++ b/nostr_dvm/tasks/videogeneration_svd.py @@ -40,7 +40,7 @@ class VideoGenerationSVD(DVMTaskInterface): request_form["trainerFilePath"] = r'modules\stablevideodiffusion\stablevideodiffusion.trainer' url = "" - frames = 14 # 25 + frames = 7 # 25 model = "stabilityai/stable-video-diffusion-img2vid-xt" #,stabilityai/stable-video-diffusion-img2vid @@ -68,7 +68,7 @@ class VideoGenerationSVD(DVMTaskInterface): options = { "model": model, - "fps": 14 + "fps": frames } request_form['options'] = json.dumps(options)