Update hunyuanvideo_v2v_6G.py

This commit is contained in:
Zhongjie Duan
2024-12-23 20:57:58 +08:00
committed by GitHub
parent edc9272e55
commit 010c801081

View File

@@ -1,5 +1,5 @@
import torch
torch.cuda.set_per_process_memory_fraction(6/80, 0)
torch.cuda.set_per_process_memory_fraction(1.0, 0)
from diffsynth import ModelManager, HunyuanVideoPipeline, download_models, save_video, FlowMatchScheduler