KenSchae
/
Pallaidium
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

Add: PixArt Sigma 2k, 1024 and RealViz V4 

NB. Dependencies needs to be updated in order to get PixArt 2K to work.
main
tin2tin 7 months ago committed by GitHub
parent
9f46426df5
commit
a10ebbcf96
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 111 additions and 50 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 129
      __init__.py

129
__init__.py
Unescape View File

@ -709,8 +709,8 @@ def install_modules(self):
else: else:
import_module(self, "resemble_enhance", "resemble-enhance") import_module(self, "resemble_enhance", "resemble-enhance")
import_module(self, "diffusers", "diffusers") #import_module(self, "diffusers", "diffusers")
#import_module(self, "diffusers", "git+https://github.com/huggingface/diffusers.git") import_module(self, "diffusers", "git+https://github.com/huggingface/diffusers.git")
subprocess.check_call([pybin, "-m", "pip", "install", "tensorflow"]) subprocess.check_call([pybin, "-m", "pip", "install", "tensorflow"])
import_module(self, "soundfile", "PySoundFile") import_module(self, "soundfile", "PySoundFile")
import_module(self, "sentencepiece", "sentencepiece") import_module(self, "sentencepiece", "sentencepiece")
@ -722,6 +722,9 @@ def install_modules(self):
import_module(self, "omegaconf", "omegaconf") import_module(self, "omegaconf", "omegaconf")
import_module(self, "protobuf", "protobuf") import_module(self, "protobuf", "protobuf")
import_module(self, "beautifulsoup4", "beautifulsoup4")
import_module(self, "ftfy", "ftfy")
python_version_info = sys.version_info python_version_info = sys.version_info
python_version_str = parse_python_version(python_version_info) python_version_str = parse_python_version(python_version_info)
@ -904,6 +907,9 @@ class GENERATOR_OT_uninstall(Operator):
uninstall_module_with_dependencies("resemble_enhance") uninstall_module_with_dependencies("resemble_enhance")
uninstall_module_with_dependencies("mediapipe") uninstall_module_with_dependencies("mediapipe")
uninstall_module_with_dependencies("beautifulsoup4")
uninstall_module_with_dependencies("ftfy")
# "resemble-enhance": # "resemble-enhance":
uninstall_module_with_dependencies("celluloid") uninstall_module_with_dependencies("celluloid")
uninstall_module_with_dependencies("omegaconf") uninstall_module_with_dependencies("omegaconf")
@ -985,7 +991,7 @@ def input_strips_updated(self, context):
scene.input_strips = "input_strips" scene.input_strips = "input_strips"
if ( if (
movie_model_card == "guoyww/animatediff-motion-adapter-v1-5-2" movie_model_card == "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta"
and type == "movie" and type == "movie"
): ):
scene.input_strips = "input_prompt" scene.input_strips = "input_prompt"
@ -1045,7 +1051,7 @@ def output_strips_updated(self, context):
scene.input_strips = "input_strips" scene.input_strips = "input_strips"
if ( if (
movie_model_card == "guoyww/animatediff-motion-adapter-v1-5-2" movie_model_card == "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta"
and type == "movie" and type == "movie"
): ):
scene.input_strips = "input_prompt" scene.input_strips = "input_prompt"
@ -1130,7 +1136,7 @@ class GeneratorAddonPreferences(AddonPreferences):
# "Zeroscope (448x256x30)", # "Zeroscope (448x256x30)",
# ), # ),
( (
"guoyww/animatediff-motion-adapter-v1-5-2", "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta",
"AnimateDiff", "AnimateDiff",
"AnimateDiff", "AnimateDiff",
), ),
@ -1183,11 +1189,21 @@ class GeneratorAddonPreferences(AddonPreferences):
"Segmind SSD-1B (1024x1024)", "Segmind SSD-1B (1024x1024)",
"segmind/SSD-1B", "segmind/SSD-1B",
), ),
# ( (
# "dataautogpt3/Miniaturus_PotentiaV1.2", "SG161222/RealVisXL_V4.0",
# "Miniaturus_PotentiaV1.2 (1024x1024)", "RealVisXL_V4 (1024x1024)",
# "dataautogpt3/Miniaturus_PotentiaV1.2", "SG161222/RealVisXL_V4.0",
# ),# ),#
(
"PixArt-alpha/PixArt-XL-2-1024-MS",
"PixArt XL (1024x1024)",
"PixArt-alpha/PixArt-XL-2-1024-MS",
),
(
"Vargol/PixArt-Sigma_2k_16bit",
"PixArt Sigma XL 2K (2560x1440)",
"Vargol/PixArt-Sigma_2k_16bit",
),
( (
"dataautogpt3/Proteus-RunDiffusion", "dataautogpt3/Proteus-RunDiffusion",
"Proteus-RunDiffusion (1024x1024)", "Proteus-RunDiffusion (1024x1024)",
@ -1195,7 +1211,7 @@ class GeneratorAddonPreferences(AddonPreferences):
), ),
("dataautogpt3/Proteus-RunDiffusion-Lightning", "ProteusV0.3-Lightning (1024 x 1024)", "dataautogpt3/Proteus-RunDiffusion-Lightning"), ("dataautogpt3/Proteus-RunDiffusion-Lightning", "ProteusV0.3-Lightning (1024 x 1024)", "dataautogpt3/Proteus-RunDiffusion-Lightning"),
("dataautogpt3/OpenDalleV1.1", "OpenDalle (1024 x 1024)", "dataautogpt3/OpenDalleV1.1"), ("dataautogpt3/OpenDalleV1.1", "OpenDalle (1024 x 1024)", "dataautogpt3/OpenDalleV1.1"),
# ("h94/IP-Adapter", "IP-Adapter (512 x 512)", "h94/IP-Adapter"), ("h94/IP-Adapter", "IP-Adapter (512 x 512)", "h94/IP-Adapter"),
#("PixArt-alpha/PixArt-XL-2-1024-MS", "PixArt (1024 x 1024)", "PixArt-alpha/PixArt-XL-2-1024-MS"), #("PixArt-alpha/PixArt-XL-2-1024-MS", "PixArt (1024 x 1024)", "PixArt-alpha/PixArt-XL-2-1024-MS"),
### ("ptx0/terminus-xl-gamma-v1", "Terminus XL Gamma v1", "ptx0/terminus-xl-gamma-v1"), ### ("ptx0/terminus-xl-gamma-v1", "Terminus XL Gamma v1", "ptx0/terminus-xl-gamma-v1"),
# ("warp-ai/wuerstchen", "Würstchen (1024x1024)", "warp-ai/wuerstchen"), # ("warp-ai/wuerstchen", "Würstchen (1024x1024)", "warp-ai/wuerstchen"),
@ -1683,7 +1699,7 @@ class SEQUENCER_PT_pallaidium_panel(Panel): # UI
if type != "audio": if type != "audio":
if ( if (
type == "movie" type == "movie"
and movie_model_card != "guoyww/animatediff-motion-adapter-v1-5-2" and movie_model_card != "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta"
) or ( ) or (
type == "image" type == "image"
#and image_model_card != "diffusers/controlnet-canny-sdxl-1.0-small" #and image_model_card != "diffusers/controlnet-canny-sdxl-1.0-small"
@ -1931,6 +1947,10 @@ class SEQUENCER_PT_pallaidium_panel(Panel): # UI
type == "image" type == "image"
and image_model_card == "PixArt-alpha/PixArt-XL-2-1024-MS" and image_model_card == "PixArt-alpha/PixArt-XL-2-1024-MS"
) )
or (
type == "image"
and image_model_card == "Vargol/PixArt-Sigma_2k_16bit"
)
): ):
row.prop(context.scene, "use_lcm", text="Speed") row.prop(context.scene, "use_lcm", text="Speed")
# Output. # Output.
@ -2060,7 +2080,7 @@ class SEQUENCER_OT_generate_movie(Operator):
print("Model: " + movie_model_card) print("Model: " + movie_model_card)
# Models for refine imported image or movie # Models for refine imported image or movie
if ((scene.movie_path or scene.image_path) and input == "input_strips" and movie_model_card != "guoyww/animatediff-motion-adapter-v1-5-2"): if ((scene.movie_path or scene.image_path) and input == "input_strips" and movie_model_card != "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta"):
if movie_model_card == "stabilityai/sd-turbo": # img2img if movie_model_card == "stabilityai/sd-turbo": # img2img
from diffusers import AutoPipelineForImage2Image from diffusers import AutoPipelineForImage2Image
@ -2218,19 +2238,17 @@ class SEQUENCER_OT_generate_movie(Operator):
# Models for movie generation # Models for movie generation
else: else:
if movie_model_card == "guoyww/animatediff-motion-adapter-v1-5-2": if movie_model_card == "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta":
from diffusers import MotionAdapter, AnimateDiffPipeline, DDIMScheduler from diffusers import MotionAdapter, AnimateDiffPipeline, DDIMScheduler
from diffusers.utils import export_to_gif from diffusers.utils import export_to_gif
# Load the motion adapter # Load the motion adapter
adapter = MotionAdapter.from_pretrained( adapter = MotionAdapter.from_pretrained(
"guoyww/animatediff-motion-adapter-v1-5-2", "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta",
local_files_only=local_files_only, local_files_only=local_files_only,
) )
# load SD 1.5 based finetuned model
# model_id = "runwayml/stable-diffusion-v1-5" model_id = "stabilityai/stable-diffusion-xl-base-1.0"
model_id = "SG161222/Realistic_Vision_V5.1_noVAE"
# model_id = "pagebrain/majicmix-realistic-v7"
pipe = AnimateDiffPipeline.from_pretrained( pipe = AnimateDiffPipeline.from_pretrained(
model_id, model_id,
motion_adapter=adapter, motion_adapter=adapter,
@ -2397,7 +2415,7 @@ class SEQUENCER_OT_generate_movie(Operator):
if ( if (
(scene.movie_path or scene.image_path) (scene.movie_path or scene.image_path)
and input == "input_strips" and input == "input_strips"
and movie_model_card != "guoyww/animatediff-motion-adapter-v1-5-2" and movie_model_card != "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta"
): ):
video_path = scene.movie_path video_path = scene.movie_path
@ -2508,7 +2526,7 @@ class SEQUENCER_OT_generate_movie(Operator):
#frames = output.frames[0] #frames = output.frames[0]
#export_to_gif(frames, "animatelcm.gif") #export_to_gif(frames, "animatelcm.gif")
elif movie_model_card != "guoyww/animatediff-motion-adapter-v1-5-2": elif movie_model_card != "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta":
if scene.movie_path: if scene.movie_path:
print("Process: Video to video") print("Process: Video to video")
if not os.path.isfile(scene.movie_path): if not os.path.isfile(scene.movie_path):
@ -2563,7 +2581,7 @@ class SEQUENCER_OT_generate_movie(Operator):
# Movie. # Movie.
else: else:
print("Generate: Video") print("Generate: Video")
if movie_model_card == "guoyww/animatediff-motion-adapter-v1-5-2": if movie_model_card == "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta":
video_frames = pipe( video_frames = pipe(
prompt=prompt, prompt=prompt,
negative_prompt=negative_prompt, negative_prompt=negative_prompt,
@ -2611,7 +2629,7 @@ class SEQUENCER_OT_generate_movie(Operator):
generator=generator, generator=generator,
).frames[0] ).frames[0]
if movie_model_card == "guoyww/animatediff-motion-adapter-v1-5-2": if movie_model_card == "a-r-r-o-w/animatediff-motion-adapter-sdxl-beta":
# from diffusers.utils import export_to_video # from diffusers.utils import export_to_video
# Move to folder. # Move to folder.
video_frames = np.array(video_frames) video_frames = np.array(video_frames)
@ -2728,7 +2746,6 @@ class SEQUENCER_OT_generate_audio(Operator):
movie_num_guidance = scene.movie_num_guidance movie_num_guidance = scene.movie_num_guidance
audio_length_in_s = scene.audio_length_in_f / (scene.render.fps / scene.render.fps_base) audio_length_in_s = scene.audio_length_in_f / (scene.render.fps / scene.render.fps_base)
pipe = None pipe = None
#try:
import torch import torch
import torchaudio import torchaudio
import scipy import scipy
@ -3391,12 +3408,19 @@ class SEQUENCER_OT_generate_image(Operator):
else: else:
from diffusers import AutoPipelineForImage2Image from diffusers import AutoPipelineForImage2Image
try:
converter = AutoPipelineForImage2Image.from_pretrained( converter = AutoPipelineForImage2Image.from_pretrained(
image_model_card, image_model_card,
torch_dtype=torch.float16, torch_dtype=torch.float16,
variant="fp16", variant="fp16",
local_files_only=local_files_only, local_files_only=local_files_only,
) )
except:
converter = AutoPipelineForImage2Image.from_pretrained(
image_model_card,
torch_dtype=torch.float16,
local_files_only=local_files_only,
)
if enabled_items and input == "input_strips" and image_model_card == "stabilityai/stable-diffusion-xl-base-1.0" and (scene.image_path or scene.movie_path) and not do_inpaint: if enabled_items and input == "input_strips" and image_model_card == "stabilityai/stable-diffusion-xl-base-1.0" and (scene.image_path or scene.movie_path) and not do_inpaint:
print("LoRAs will be ignored for image or movie input.") print("LoRAs will be ignored for image or movie input.")
@ -3894,7 +3918,22 @@ class SEQUENCER_OT_generate_image(Operator):
torch_dtype=torch.float16, torch_dtype=torch.float16,
local_files_only=local_files_only, local_files_only=local_files_only,
) )
if low_vram():
pipe.enable_model_cpu_offload()
else:
pipe.to(gfx_device)
elif image_model_card == "Vargol/PixArt-Sigma_2k_16bit":
from diffusers import PixArtSigmaPipeline
pipe = PixArtSigmaPipeline.from_pretrained(
"Vargol/PixArt-Sigma_2k_16bit",
torch_dtype=torch.float16,
variant="fp16",
local_files_only=local_files_only,
)
if low_vram():
pipe.enable_model_cpu_offload()
else:
pipe.to(gfx_device)
elif image_model_card == "ByteDance/SDXL-Lightning": elif image_model_card == "ByteDance/SDXL-Lightning":
import torch import torch
from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler, AutoencoderKL from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler, AutoencoderKL
@ -3963,19 +4002,28 @@ class SEQUENCER_OT_generate_image(Operator):
# decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to(gfx_device) # decoder = StableCascadeDecoderPipeline.from_pretrained("stabilityai/stable-cascade", torch_dtype=torch.float16).to(gfx_device)
elif image_model_card == "dataautogpt3/Miniaturus_PotentiaV1.2": # elif image_model_card == "SG161222/RealVisXL_V4.0":
# from diffusers import AutoPipelineForText2Image
# pipe = AutoPipelineForText2Image.from_pretrained(
# "SG161222/RealVisXL_V4.0",
# torch_dtype=torch.float16, # vae=vae,
# variant="fp16",
# local_files_only=local_files_only,
# )
else:
try:
from diffusers import AutoPipelineForText2Image from diffusers import AutoPipelineForText2Image
pipe = AutoPipelineForText2Image.from_pretrained( pipe = AutoPipelineForText2Image.from_pretrained(
"dataautogpt3/Miniaturus_PotentiaV1.2", image_model_card,
torch_dtype=torch.float16, # vae=vae, torch_dtype=torch.float16,
variant="fp16",
local_files_only=local_files_only, local_files_only=local_files_only,
) )
else: except:
from diffusers import AutoPipelineForText2Image from diffusers import AutoPipelineForText2Image
pipe = AutoPipelineForText2Image.from_pretrained( pipe = AutoPipelineForText2Image.from_pretrained(
image_model_card, image_model_card,
torch_dtype=torch.float16, torch_dtype=torch.float16,
variant="fp16",
local_files_only=local_files_only, local_files_only=local_files_only,
) )
@ -4028,12 +4076,12 @@ class SEQUENCER_OT_generate_image(Operator):
from diffusers import EulerAncestralDiscreteScheduler from diffusers import EulerAncestralDiscreteScheduler
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config) pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
elif image_model_card != "PixArt-alpha/PixArt-XL-2-1024-MS" and image_model_card != "stabilityai/stable-cascade": elif image_model_card != "PixArt-alpha/PixArt-XL-2-1024-MS" and image_model_card != "stabilityai/stable-cascade" and image_model_card != "Vargol/PixArt-Sigma_2k_16bit":
pipe.scheduler = DPMSolverMultistepScheduler.from_config( pipe.scheduler = DPMSolverMultistepScheduler.from_config(
pipe.scheduler.config pipe.scheduler.config
) )
if image_model_card != "stabilityai/stable-cascade": if image_model_card != "stabilityai/stable-cascade" and image_model_card != "Vargol/PixArt-Sigma_2k_16bit":
pipe.watermark = NoWatermark() pipe.watermark = NoWatermark()
if low_vram(): if low_vram():
@ -4518,6 +4566,19 @@ class SEQUENCER_OT_generate_image(Operator):
generator=generator, generator=generator,
).images[0] ).images[0]
elif image_model_card == "Vargol/PixArt-Sigma_2k_16bit":
image = pipe(
# prompt_embeds=prompt, # for compel - long prompts
prompt,
negative_prompt=negative_prompt,
num_inference_steps=image_num_inference_steps,
guidance_scale=image_num_guidance,
height=y,
width=x,
generator=generator,
).images[0]
# Inpaint # Inpaint
elif do_inpaint: elif do_inpaint:
print("Process: Inpaint") print("Process: Inpaint")
@ -5262,7 +5323,7 @@ def register():
default=1024, default=1024,
step=64, step=64,
min=256, min=256,
max=1536, max=2560,
description="Use the power of 64", description="Use the power of 64",
) )
bpy.types.Scene.generate_movie_y = bpy.props.IntProperty( bpy.types.Scene.generate_movie_y = bpy.props.IntProperty(
@ -5270,7 +5331,7 @@ def register():
default=576, default=576,
step=64, step=64,
min=256, min=256,
max=1536, max=1440,
description="Use the power of 64", description="Use the power of 64",
) )
# The number of frames to be generated. # The number of frames to be generated.

Write Preview
Loading…
Cancel
Save