Update README.md

README.md

@@ -104,7 +104,7 @@ high contrast, photorealistic, intimate, elegant, visually balanced, serene atmo
 
 # Load the VisualClozePipeline
 pipe = VisualClozePipeline.from_pretrained("VisualCloze/VisualClozePipeline-384", torch_dtype=torch.bfloat16)
-pipe.enable_model_cpu_offload()  # Save some VRAM by offloading the model to CPU
+pipe.to("cuda")
 
 # Run the pipeline
 image_result = pipe(
@@ -158,7 +158,7 @@ content_prompt = None
 
 # Load the VisualClozePipeline
 pipe = VisualClozePipeline.from_pretrained("VisualCloze/VisualClozePipeline-384", torch_dtype=torch.bfloat16)
-pipe.enable_model_cpu_offload()  # Save some VRAM by offloading the model to CPU
+pipe.to("cuda")
 
 # Run the pipeline
 image_result = pipe(