Skip to content

Commit 38eaa73

Browse files
authored
add gpu memory of largedit_3b (#1249) (#1273)
1 parent fb3d4a2 commit 38eaa73

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ppdiffusers/examples/inference/text_to_image_generation_largedit_3b.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
from ppdiffusers import DPMSolverMultistepScheduler
1919
from ppdiffusers.pipelines import LDMTextToImageLargeDiTPipeline
2020

21+
# note that: will cost about 36G gpu memory
2122
dtype = paddle.bfloat16
2223
pipe = LDMTextToImageLargeDiTPipeline.from_pretrained("Alpha-VLLM/Large-DiT-T2I-3B-1024", paddle_dtype=dtype)
2324
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)

0 commit comments

Comments
 (0)