Browse Source

Update __init__.py

pull/25/head
tin2tin 1 year ago committed by GitHub
parent
commit
8aeef4d061
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 10
      __init__.py

10
__init__.py

@ -312,7 +312,7 @@ def install_modules(self):
import_module(self, "sox", "sox") import_module(self, "sox", "sox")
else: else:
import_module(self, "soundfile", "PySoundFile") import_module(self, "soundfile", "PySoundFile")
import_module(self, "diffusers", "git+https://github.com/huggingface/diffusers.git") import_module(self, "diffusers", "diffusers") #git+https://github.com/huggingface/diffusers.git")
import_module(self, "accelerate", "accelerate") import_module(self, "accelerate", "accelerate")
import_module(self, "transformers", "transformers") import_module(self, "transformers", "transformers")
import_module(self, "sentencepiece", "sentencepiece") import_module(self, "sentencepiece", "sentencepiece")
@ -422,8 +422,10 @@ class GeneratorAddonPreferences(AddonPreferences):
items=[ items=[
("strangeman3107/animov-0.1.1", "Animov (448x384)", "Animov (448x384)"), ("strangeman3107/animov-0.1.1", "Animov (448x384)", "Animov (448x384)"),
("strangeman3107/animov-512x", "Animov (512x512)", "Animov (512x512)"), ("strangeman3107/animov-512x", "Animov (512x512)", "Animov (512x512)"),
("cerspense/zeroscope_v1_320s", "Zeroscope v1 (320x320)", "Modelscope (320x320)"), ("cerspense/zeroscope_v1-1_320s", "Zeroscope v1.1 (320x320)", "Zerocope (320x320)"),
("camenduru/potat1", "Potat v1 (1024x576)", "Modelscope (1024x576)"), ("camenduru/potat1", "Potat v1 (1024x576)", "Potat (1024x576)"),
("polyware-ai/text-to-video-ms-stable-v1", "text-to-video-ms-1.7b (384x384)", "text-to-video-ms-1.7b (384x384)"),
#("vdo/potat1-50000", "Potat v1 50000 (1024x576)", "Potat (1024x576)"),
("damo-vilab/text-to-video-ms-1.7b", "Modelscope (256x256)", "Modelscope (256x256)"), ("damo-vilab/text-to-video-ms-1.7b", "Modelscope (256x256)", "Modelscope (256x256)"),
], ],
default="strangeman3107/animov-0.1.1", default="strangeman3107/animov-0.1.1",
@ -747,7 +749,7 @@ class SEQUENCER_OT_generate_movie(Operator):
# memory optimization # memory optimization
pipe.enable_model_cpu_offload() pipe.enable_model_cpu_offload()
pipe.enable_vae_slicing() pipe.enable_vae_slicing()
pipe.enable_xformers_memory_efficient_attention() #pipe.enable_xformers_memory_efficient_attention()
for i in range(scene.movie_num_batch): for i in range(scene.movie_num_batch):
#wm.progress_update(i) #wm.progress_update(i)

Loading…
Cancel
Save