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")
else:
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, "transformers", "transformers")
import_module(self, "sentencepiece", "sentencepiece")
@ -422,8 +422,10 @@ class GeneratorAddonPreferences(AddonPreferences):
items=[
("strangeman3107/animov-0.1.1", "Animov (448x384)", "Animov (448x384)"),
("strangeman3107/animov-512x", "Animov (512x512)", "Animov (512x512)"),
("cerspense/zeroscope_v1_320s", "Zeroscope v1 (320x320)", "Modelscope (320x320)"),
("camenduru/potat1", "Potat v1 (1024x576)", "Modelscope (1024x576)"),
("cerspense/zeroscope_v1-1_320s", "Zeroscope v1.1 (320x320)", "Zerocope (320x320)"),
("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)"),
],
default="strangeman3107/animov-0.1.1",
@ -747,7 +749,7 @@ class SEQUENCER_OT_generate_movie(Operator):
# memory optimization
pipe.enable_model_cpu_offload()
pipe.enable_vae_slicing()
pipe.enable_xformers_memory_efficient_attention()
#pipe.enable_xformers_memory_efficient_attention()
for i in range(scene.movie_num_batch):
#wm.progress_update(i)

Loading…
Cancel
Save