Update gradio_sd3.py
Browse files- gradio_sd3.py +1 -0
gradio_sd3.py
CHANGED
|
@@ -38,6 +38,7 @@ pipeline = StableDiffusion3TryOnPipeline.from_pretrained(repo_path, torch_dtype=
|
|
| 38 |
transformer_garm=transformer_garm, transformer_vton=transformer_vton, pose_guider=pose_guider, \
|
| 39 |
image_encoder_large=image_encoder_large, image_encoder_bigG=image_encoder_bigG)
|
| 40 |
pipeline.to(device)
|
|
|
|
| 41 |
dwprocessor = DWposeDetector(model_root=repo_path, device=device)
|
| 42 |
parsing_model = Parsing(model_root=repo_path, device=device)
|
| 43 |
|
|
|
|
| 38 |
transformer_garm=transformer_garm, transformer_vton=transformer_vton, pose_guider=pose_guider, \
|
| 39 |
image_encoder_large=image_encoder_large, image_encoder_bigG=image_encoder_bigG)
|
| 40 |
pipeline.to(device)
|
| 41 |
+
self.pipeline.enable_model_cpu_offload()
|
| 42 |
dwprocessor = DWposeDetector(model_root=repo_path, device=device)
|
| 43 |
parsing_model = Parsing(model_root=repo_path, device=device)
|
| 44 |
|