|
|
|
@ -673,15 +673,17 @@ def install_modules(self):
|
|
|
|
|
|
|
|
|
|
import_module(self, "huggingface_hub", "huggingface_hub") |
|
|
|
|
import_module(self, "accelerate", "git+https://github.com/huggingface/accelerate.git") |
|
|
|
|
#import_module(self, "transformers", "git+https://github.com/huggingface/transformers") |
|
|
|
|
subprocess.check_call([pybin, "-m", "pip", "install", "transformers", "--upgrade"]) |
|
|
|
|
#import_module(self, "transformers", "git+https://github.com/huggingface/transformers.git") |
|
|
|
|
subprocess.check_call([pybin, "-m", "pip", "install", "git+https://github.com/huggingface/transformers.git", "--upgrade"]) |
|
|
|
|
#subprocess.check_call([pybin, "-m", "pip", "install", "transformers", "--upgrade"]) |
|
|
|
|
import_module(self, "bark", "git+https://github.com/suno-ai/bark.git") |
|
|
|
|
#import_module(self, "bark", "git+https://github.com/suno-ai/bark.git") |
|
|
|
|
import_module(self, "diffusers", "diffusers") |
|
|
|
|
#import_module(self, "diffusers", "git+https://github.com/huggingface/diffusers.git@v0.22.3") |
|
|
|
|
import_module(self, "tensorflow", "tensorflow") |
|
|
|
|
if os_platform == "Darwin" or os_platform == "Linux": |
|
|
|
|
import_module(self, "sox", "sox") |
|
|
|
|
else: |
|
|
|
|
# if os_platform == "Darwin" or os_platform == "Linux": |
|
|
|
|
# import_module(self, "sox", "sox") |
|
|
|
|
# else: |
|
|
|
|
import_module(self, "soundfile", "PySoundFile") |
|
|
|
|
#import_module(self, "transformers", "transformers") |
|
|
|
|
import_module(self, "sentencepiece", "sentencepiece") |
|
|
|
@ -880,9 +882,9 @@ class GENERATOR_OT_uninstall(Operator):
|
|
|
|
|
uninstall_module_with_dependencies("torchvision") |
|
|
|
|
uninstall_module_with_dependencies("torchaudio") |
|
|
|
|
|
|
|
|
|
if os_platform == "Darwin" or os_platform == "Linux": |
|
|
|
|
uninstall_module_with_dependencies("sox") |
|
|
|
|
else: |
|
|
|
|
# if os_platform == "Darwin" or os_platform == "Linux": |
|
|
|
|
# uninstall_module_with_dependencies("sox") |
|
|
|
|
# else: |
|
|
|
|
uninstall_module_with_dependencies("PySoundFile") |
|
|
|
|
uninstall_module_with_dependencies("diffusers") |
|
|
|
|
uninstall_module_with_dependencies("accelerate") |
|
|
|
@ -1861,6 +1863,7 @@ class SEQUENCER_OT_generate_movie(Operator):
|
|
|
|
|
|
|
|
|
|
vae = AutoencoderKL.from_pretrained( |
|
|
|
|
"madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16 |
|
|
|
|
|
|
|
|
|
) |
|
|
|
|
pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained( |
|
|
|
|
movie_model_card, |
|
|
|
@ -1997,6 +2000,7 @@ class SEQUENCER_OT_generate_movie(Operator):
|
|
|
|
|
else: |
|
|
|
|
|
|
|
|
|
from diffusers import TextToVideoSDPipeline |
|
|
|
|
import torch |
|
|
|
|
|
|
|
|
|
pipe = TextToVideoSDPipeline.from_pretrained( |
|
|
|
|
movie_model_card, |
|
|
|
@ -2367,9 +2371,9 @@ class SEQUENCER_OT_generate_audio(Operator):
|
|
|
|
|
import xformers |
|
|
|
|
|
|
|
|
|
if addon_prefs.audio_model_card == "facebook/musicgen-stereo-small": |
|
|
|
|
if os_platform == "Darwin" or os_platform == "Linux": |
|
|
|
|
import sox |
|
|
|
|
else: |
|
|
|
|
# if os_platform == "Darwin" or os_platform == "Linux": |
|
|
|
|
# import sox |
|
|
|
|
# else: |
|
|
|
|
import soundfile as sf |
|
|
|
|
|
|
|
|
|
if addon_prefs.audio_model_card == "bark": |
|
|
|
@ -2519,14 +2523,14 @@ class SEQUENCER_OT_generate_audio(Operator):
|
|
|
|
|
filename = solve_path(clean_filename(str(seed)+"_"+prompt) + ".wav") |
|
|
|
|
rate = 48000 |
|
|
|
|
|
|
|
|
|
if os_platform == "Darwin" or os_platform == "Linux": |
|
|
|
|
tfm = sox.Transformer() |
|
|
|
|
tfm.build_file( |
|
|
|
|
input_array=music["audio"][0].T, |
|
|
|
|
sample_rate_in=music["sampling_rate"], |
|
|
|
|
output_filepath=filename |
|
|
|
|
) |
|
|
|
|
else: |
|
|
|
|
# if os_platform == "Darwin" or os_platform == "Linux": |
|
|
|
|
# tfm = sox.Transformer() |
|
|
|
|
# tfm.build_file( |
|
|
|
|
# input_array=music["audio"][0].T, |
|
|
|
|
# sample_rate_in=music["sampling_rate"], |
|
|
|
|
# output_filepath=filename |
|
|
|
|
# ) |
|
|
|
|
# else: |
|
|
|
|
sf.write(filename, music["audio"][0].T, music["sampling_rate"]) |
|
|
|
|
|
|
|
|
|
elif addon_prefs.audio_model_card == "declare-lab/mustango": |
|
|
|
|