Browse Source

Downgrade torch if necessary

pull/17/head
tin2tin 2 years ago committed by GitHub
parent
commit
43009c70d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 23
      __init__.py

23
__init__.py

@ -281,7 +281,7 @@ def install_modules(self):
import_module(self, "torch", "torch") import_module(self, "torch", "torch")
import_module(self, "torchvision", "torchvision") import_module(self, "torchvision", "torchvision")
import_module(self, "torchaudio", "torchaudio") import_module(self, "torchaudio", "torchaudio")
if os_platform == 'Darwin': if os_platform == 'Darwin' or os_platform == 'Linux':
import_module(self, "sox", "sox") import_module(self, "sox", "sox")
else: else:
import_module(self, "soundfile", "PySoundFile") import_module(self, "soundfile", "PySoundFile")
@ -296,6 +296,21 @@ def install_modules(self):
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, "IPython", "IPython") import_module(self, "IPython", "IPython")
subprocess.check_call([pybin,"-m","pip","install","numpy","--upgrade"]) subprocess.check_call([pybin,"-m","pip","install","numpy","--upgrade"])
if os_platform == "Windows":
subprocess.check_call(
[
pybin,
"-m",
"pip",
"install",
"torch",
"--index-url",
"https://download.pytorch.org/whl/cu118",
"--no-warn-script-location",
"--user",
]
)
class GeneratorAddonPreferences(AddonPreferences): class GeneratorAddonPreferences(AddonPreferences):
@ -332,9 +347,9 @@ class GeneratorAddonPreferences(AddonPreferences):
movie_model_card: bpy.props.EnumProperty( movie_model_card: bpy.props.EnumProperty(
name="Movie Model Card", name="Movie Model Card",
items=[ items=[
("strangeman3107/animov-0.1.1", "Animov (448x384)", "Animov"), ("strangeman3107/animov-0.1.1", "Animov (448x384)", "Animov (448x384)"),
("strangeman3107/animov-512x", "Animov (512x512)", "Animov"), ("strangeman3107/animov-512x", "Animov (512x512)", "Animov (512x512)"),
("damo-vilab/text-to-video-ms-1.7b", "Modelscope (256x256)", "Modelscope"), ("damo-vilab/text-to-video-ms-1.7b", "Modelscope (256x256)", "Modelscope (256x256)"),
], ],
default="strangeman3107/animov-0.1.1", default="strangeman3107/animov-0.1.1",
) )

Loading…
Cancel
Save