Starting the cogview3-plus model diffusers/models/modeling_utils.py line 774 yields an error TypeError: load_checkpoint_and_dispatch() got an unexpected keyword argument 'strict'. Line 783 exegesis 'strict=True,' can start the model
Reproduction
from diffusers import CogView3PlusPipeline
import torch
pipe = CogView3PlusPipeline.from_pretrained("F:/CogView3-Plus-3B", torch_dtype=torch.bfloat16).to("cuda")
# Open it for reduce GPU memory usage
# pipe.enable_model_cpu_offload()
# pipe.vae.enable_slicing()
# pipe.vae.enable_tiling()
prompt = "A vibrant cherry red sports car sits proudly under the gleaming sun, its polished exterior smooth and flawless, casting a mirror-like reflection. The car features a low, aerodynamic body, angular headlights that gaze forward like predatory eyes, and a set of black, high-gloss racing rims that contrast starkly with the red. A subtle hint of chrome embellishes the grille and exhaust, while the tinted windows suggest a luxurious and private interior. The scene conveys a sense of speed and elegance, the car appearing as if it's about to burst into a sprint along a coastal road, with the ocean's azure waves crashing in the background."
image = pipe(
prompt=prompt,
guidance_scale=7.0,
num_images_per_prompt=1,
num_inference_steps=1,
width=512,
height=512,
).images[0]
image.save("F:/cogview3.png")
###
Describe the bugStarting the cogview3-plus model diffusers/models/modeling_utils.py line 774 yields an error TypeError: load_checkpoint_and_dispatch() got an unexpected keyword argument 'strict'. Line 783 exegesis 'strict=True,' can start the model
Reproduction
Logs
No response
System Info
diffusers-0.31.0.dev0
Who can help?
No response