BASSEM45325

i have tried and got error pipe = CogVideoXImageToVideoPipeline.from_pretrained( "THUDM/CogVideoX-5b-I2V", torch_dtype=torch.bfloat16, device_map="balanced" ) pipe.scheduler = CogVideoXDPMScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing") pipe.vae.enable_tiling() pipe.vae.enable_slicing() video = pipe( prompt=prompt, image=image, num_videos_per_prompt=1, num_inference_steps=50, num_frames=49, guidance_scale=6, use_dynamic_cfg=True, generator=torch.Generator().manual_seed(112),...

it solved when i used this i dont know why it not working on 4 gpus it only working on 2 gpus os.environ["CUDA_VISIBLE_DEVICES"] = "0,1

and when i put .to('cuda') igot error also