HVision-NKU / StoryDiffusion

Accepted as [NeurIPS 2024] Spotlight Presentation Paper
Apache License 2.0
5.83k stars 582 forks source link

can this work on Mac M3 ? how much ram required for that ? #42

Closed scottonly2 closed 4 months ago

kellyzxiaowei commented 5 months ago

+1

kellyzxiaowei commented 4 months ago

I made a modification that can run on the M series chip, and I tested it on the M3 Max with 64GB, where it worked normally. gradio_app_sdxl_specific_id_mps.py.zip

Z-YuPeng commented 4 months ago

@kellyzxiaowei Could you make a pull request about it? Very thankful for your efforts.

kellyzxiaowei commented 4 months ago

@kellyzxiaowei Could you make a pull request about it? Very thankful for your efforts. I have already submitted a pull request.

scottonly2 commented 4 months ago

How much RAM required to use the mps version ? will 16G enough , and is it possible to adjust with some settings

kellyzxiaowei commented 4 months ago

How much RAM required to use the mps version ? will 16G enough , and is it possible to adjust with some settings

The peak memory usage is 45.7 GB, which makes me think that 16 GB may not work properly. Physical footprint: 45.5G Physical footprint (peak): 45.7G Idle exit: untracked

scottonly2 commented 4 months ago

Hi kelly, can you check the error I report when run with mps , I tried with 16G RAM M3, with PYTORCH_MPS_HIGH_WATERMARK_RATIO=0.5