HVision-NKU / StoryDiffusion

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

OSError pytorch #64

Open yolichek opened 4 months ago

yolichek commented 4 months ago

When run via jupyter :

WARNING[XFORMERS]: xFormers can't load C++/CUDA extensions. xFormers was built for: PyTorch 2.0.1+cu118 with CUDA 1108 (you have 2.0.1+cpu) Python 3.10.11 (you have 3.10.14) Please reinstall xformers (see https://github.com/facebookresearch/xformers#installing-xformers) Memory-efficient attention, SwiGLU, sparse and more won't be available. Set XFORMERS_MORE_DETAILS=1 for more details

torch.cuda.is_available() False

OSError: Error no file named pytorch_model.bin, tf_model.h5, model.ckpt.index or flax_model.msgpack found in directory C:\Users...

HowlingNitin commented 4 months ago
  1. Reinstall xFormers:

    • Follow the instructions provided in the warning message:
      pip uninstall xformers
      pip install xformers --extra-index-url https://download.pytorch.org/whl/cu118
    • This will reinstall xFormers with the correct CUDA support for your environment (assuming your GPU is compatible with CUDA 11.8).

    If your GPU doesn't support CUDA 11.8:

    • Check the xFormers documentation for compatible CUDA versions and reinstall with the appropriate version using the --extra-index-url option.
    • Consider upgrading your GPU drivers if necessary.
  2. Locate or Provide the Model File:

    • Make sure the model file you're trying to load exists in the specified directory (C:\Users...). Double-check the path and filename.
    • If the model file isn't generated by your code, ensure you have downloaded it from the correct source.

Additional Tips: