cumulo-autumn / StreamDiffusion

StreamDiffusion: A Pipeline-Level Solution for Real-Time Interactive Generation
Apache License 2.0
9.48k stars 677 forks source link

trouble installing on Mac #134

Open dfl opened 6 months ago

dfl commented 6 months ago

I had trouble installing on my Mac, but here's what I did to get it to work.

git clone https://github.com/cumulo-autumn/StreamDiffusion.git cd StreamDiffusion pip install . git patch apply ignore_cuda_for_mac.patch

brentkearney commented 6 months ago

For me, it's just: git apply ignore_cuda_for_mac.patch. Thanks for the tip!

But then what? Installation Steps 2 & 3 are to pick your Cuda version for PyTorch, and then Install TensorRT extension:

% python -m streamdiffusion.tools.install-tensorrt
Traceback (most recent call last):
  File "<frozen runpy>", line 198, in _run_module_as_main
  File "<frozen runpy>", line 88, in _run_code
  File "/Users/brent/anaconda3/lib/python3.11/site-packages/streamdiffusion/tools/install-tensorrt.py", line 19, in <module>
    def install(cu: Optional[Literal["11", "12"]] = get_cuda_version_from_torch()):
                                                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/Users/brent/anaconda3/lib/python3.11/site-packages/streamdiffusion/tools/install-tensorrt.py", line 16, in get_cuda_version_from_torch
    return torch.version.cuda.split(".")[0]
           ^^^^^^^^^^^^^^^^^^^^^^^^
AttributeError: 'NoneType' object has no attribute 'split'
fbarretto commented 4 months ago

I've created this gist to help guide the setup and running the demos. Let me know if this works for you.