Pytorch directml stable diffusion download


1-768. Fooocus. Stable Diffusion WebUI Forge is a platform on top of Stable Diffusion WebUI (based on Gradio) to make development easier, optimize resource management, speed up inference, and study experimental features. (As of 1/15/23 you can just run webui. diffusion. Download the weights for Stable Diffusion. org, along with instructions for local installation in the same simple, selectable format as PyTorch packages for CPU-only configurations and other GPU platforms. Jul 17, 2023 · C:\Users\alias\stable-diffusion-webui-directml>call webui. Follow the ComfyUI manual installation instructions for Windows and Linux. If there isn’t an ONNX model branch available, use the main branch and convert it to ONNX. Navigate to the examples\inference folder, there should be a file named save_onnx. It supports basic and ControlNet enhanced implementations of txt2img, img2img, inpainting pipelines and the safety checker. DirectML is already pre-installed on a huge range of Windows 10 Installation breaks because you don't have CUDA toolkit (if you don't have NVIDIA GPU). After a few months of community efforts, Intel Arc finally has its own Stable Diffusion Web UI! There are currently 2 available versions - one relies on DirectML and one relies on oneAPI, the latter of which is a comparably faster implementation and uses less VRAM for Arc despite being in its infant stage. Next. We should wait for the next update on torch-directml that supports PyTorch 2. 5mins for cpu and 8mins for GPU. 👍 1 RaiMelken reacted with thumbs up emoji ️ 1 RaiMelken reacted with heart emoji . The issue has been reported before but has Nov 15, 2022 · During handling of the above exception, another exception occurred: Traceback (most recent call last): File "D:\STABLE DIFFUSION\stable-diffusion-webui\venv\lib\site-packages\pip_internal\cli\base_command. In the System Properties window, click “Environment Variables. 0. /webui. Downloads always resume when possible. x, SD2. Stable represents the most currently tested and supported version of PyTorch. Install pytorch nightly. PyTorch-DirectML now works with Python versions 3. To check the optimized model, you can type: python stable_diffusion. sh shell script in the root folder, then retry running the webui-user. 1, Hugging Face) at 768x768 resolution, based on SD2. Yet another PyTorch implementation of Stable Diffusion. Getting Started See the Plugin Installation Guide for instructions. PyTorch on ROCm includes full Nov 15, 2023 · Olive is a powerful open-source Microsoft tool to optimize ONNX models for DirectML. Transformer graph optimization: fuses subgraphs into multi-head attention operators and eliminating inefficient from conversion. With ONNX Runtime, developers can extend their Windows applications to other platforms like web, cloud or mobile, wherever they need to ship their application on. re-ran webui-user. Loading Guides for how to load and configure all the components (pipelines, models, and schedulers) of the library, as well as how to use different schedulers. Fully supports SD1. DirectML for PyTorch is version 0. add altdiffusion-m18 support ( #13364) support inference with LyCORIS GLora networks ( #13610) add lora-embedding bundle system ( #13568) option to move prompt from top row into generation parameters. Nov 30, 2023 · The DirectML sample for Stable Diffusion applies the following techniques: Model conversion: translates the base models from PyTorch to ONNX. internal '. Select your preferences and run the install command. We didn’t want to stop there, since many users access Stable Diffusion through Automatic1111' I have a rx6500xt and i5 11400F. Stable Diffusion is a latent diffusion model conditioned on the (non-pooled) text embeddings of a CLIP ViT-L/14 text encoder. This Python script will convert the Stable Diffusion model into onnx files. This approach uses less video memory, generates larger images, and reduces the whine of Intel graphics cards during processing. Sep 11, 2023 · First, you need to set up your development environment as is explained in the installation section. 0, XT 1. if you are a Windows user you can try the DirectML Download the stable-diffusion Feb 1, 2024 · This can happen if your PyTorch and torchvision versions are incompatible, or if you had errors while compiling torchvision from source. The above exception was the direct cause of the following exception: Traceback (most recent call last): File "H:\Instalan Game\AI Stable Defusion\stable-diffusion-webui-directml\venv\lib\site-packages\transformers\utils\import_utils. News. Sep 22, 2022 · delete the venv directory (wherever you cloned the stable-diffusion-webui, e. This concludes our Environment build for Stable Diffusion on an AMD GPU on D:\GitResource\stable-diffusion-webui-directml\venv\lib\site-packages\pytorch_lightning\utilities\distributed. Option 2: Use the 64-bit Windows installer provided by the Python website. This step will take a few minutes depending on your CPU speed. Fooocus is a rethinking of Stable Diffusion and Midjourney’s designs: Learned from Stable Diffusion, the software is offline, open source, and free. This should be suitable for many users. 2+cu121. python save_onnx. May 23, 2023 · Stable Diffusion models with different checkpoints and/or weights but the same architecture and layers as these models will work well with Olive. py. Largely depends on practical performance (the previous DirectML iterations were slow as shit no matter the hardware; like, better than using a CPU, but not by that much) and actual compatibility (supporting Pytorch is good, but does it support all of pytorch or will it break half the time like the other times AMD DirectML/OpenCL has been "supporting" something and just weren't compatible, and I've downloaded the Stable-Diffusion-WebUI-DirectML, the k-diffusion and Stability-AI's stablediffusion Extensions, also. Aloereed/stable-diffusion-webui-ipex-arc . Fully portable - move Stability Matrix's Data Directory to a new drive or computer at any Oct 5, 2022 · Dual booted to EndeavourOS (Arch) and Stable Diffusion Native Isekai Too Guide using the arch4edu ROCm pytorch. If you used the environment file above to set up Conda, choose the `cp39` file (aka Python 3. Configs are hard-coded (based on Stable Diffusion v1. x, SDXL, Stable Video Diffusion, Stable Cascade and SD3; Asynchronous Queue system; Many optimizations: Only re-executes the parts of the workflow that changes between executions. 8. No token limit for prompts (original stable diffusion lets you use up to 75 tokens) DeepDanbooru integration, creates danbooru style tags for anime prompts xformers , major speed increase for select cards: (add --xformers to commandline args) Install and run with:. If you have another Stable Diffusion UI you might be able to reuse the No token limit for prompts (original stable diffusion lets you use up to 75 tokens) DeepDanbooru integration, creates danbooru style tags for anime prompts xformers , major speed increase for select cards: (add --xformers to commandline args) Jan 16, 2024 · Option 1: Install from the Microsoft store. Stable Diffusion versions 1. Please consider migrating to SD. The issue is caused by an extension, but I believe it is caused by a bug in the webui. May 24, 2022 · Project description. supported Features: text2img, img2img, Inpaint, textual inversion, Lora/LoHa/LyCORIS, Controlnet, Upscaling, models converter, custom vae This repository contains a fully C++ implementation of Stable Diffusion-based image synthesis tool called Unpaint. Sep 11, 2023 · Model Description: This is a trained model based on SDXL that can be used to generate and modify images based on text prompts. File "C:\Users\Arm\sd\stable-diffusion-webui-directml\modules\launch_utils. Nodes/graph/flowchart interface to experiment and create complex Stable Diffusion workflows without needing to code anything. bat, it always pops out No module 'xformers'. Then, be prepared to WAIT for that first model load Stable Diffusion in pure C/C++. ”. 4; Stable Diffusion Models v1. New stable diffusion finetune (Stable unCLIP 2. First, remove all Python versions you have previously installed. git clone https://github. Start Locally. 0-base and the Pokemons dataset. Learned from Midjourney, the manual tweaking is not needed, and users only need to focus on the prompts and images. Feb 8, 2023 · I've downgraded to 3. I tried at least this 1. combine these 2 changes, it works perfectly fine. dml = torch_directml. installation files and folders connected with SD, if you don't need it, uninstall all other versions of python as you will probably need to manually route SD to 3. -run the "stable-diffusion-webui-directml" folder on the command line as an administrator Feb 25, 2022 · Today, we are releasing the Second Preview with significant performance improvements and greater coverage for computer vision models. Without further ado let's get into how May 5, 2023 · Maybe you could try to install PyTorch manually using our install instructions in another virtual environment to check if it can detect and use your GPU. It is a Latent Diffusion Model that uses two fixed, pretrained text encoders ( OpenCLIP-ViT/G and CLIP-ViT/L ). Getting 2. py:1132: FutureWarning: ` resume_download ` is deprecated and will be removed in version 1. Attention mask at CLIP tokenizer/encoder). Aug 19, 2023 · from ldm. 10). 06. 13. bat. [2023. See here for a Python sample This repository contains a conversion tool, some examples, and instructions on how to set up Stable Diffusion with ONNX models. Stable UnCLIP 2. 0 pre-release. 10-venv -y. PyTorch is a Python package that provides two high-level features: Tensor computation (like NumPy) with strong GPU acceleration. C:\Users\paulw\STABLEDIFFUSION\WEBUI\stable-diffusion-webui-directml\venv\lib\site-packages\pytorch_lightning\utilities\distributed. I've documented the procedure I used to get Stable Diffusion up and running on my AMD Radeon 6800XT card. C:\Users\you\stable-diffusion-webui\venv) check the environment variables (click the Start button, then type “environment properties” into the search bar and hit Enter. This method should work for all the newer navi cards that are supported by ROCm. The Windows AI team is excited to announce the first preview of DirectML as a backend to PyTorch for training ML models! This release is our first step towards unlocking accelerated machine learning training for PyTorch on any DirectX12 GPU on Windows and the Windows Subsystem for Linux (WSL). May 26, 2024 · Creating model from config: C: \U sers \~\s table-diffusion-webui-directml \c onfigs \v 1-inference. py", line 20, in <module> from pytorch_lightning. status = run_func (*args) File "D:\STABLE DIFFUSION\stable-diffusion-webui\venv\lib\site-packages\pip We will leverage and download the ONNX Stable Diffusion models from Hugging Face. If --upcast-sampling works as a fix with your card, you should have 2x speed (fp16) compared to running in full precisi Stable Diffusion web UI for Intel Arc with Intel Extension for Pytorch. 10 to PATH “) I recommend installing it from the Microsoft store. This solution does not depend on Python and runs the entire image generation process in a single process Good luck you peasants, green is the way to go for gpus. Check out tomorrow’s Build Breakout Session to see Stable Diffusion in action: Deliver AI-powered experiences across cloud and edge, with Windows. Stable Diffusion WebUI Forge. 9). exe" WARNING: ZLUDA works best with SD. I want to present our UI for SD: It positioned as simple UI for SD funs. py in def prepare_environemnt(): function add xformers to commandline_ar May 23, 2023 · Stable Diffusion is a text-to-image model that transforms natural language into stunning images. import torch. To reproduce, setup environment: pip install virtualenv python -m venv sd_env_torchdml sd_env_to Download the WHL file for your Python environment. Installing ComfyUI: Mar 3, 2024 · Checklist. Once this is done you could check if the mention application could reuse your PyTorch installation instead of installing a new one. bat - it installs the new xformers that works with PyTorch 2. The issue exists on a clean installation of webui. Saved searches Use saved searches to filter your results more quickly A basic crash course for learning how to use the library's most important features like using models and schedulers to build your own diffusion system, and training your own diffusion model. To Test the Optimized Model Sep 8, 2023 · I'm trying to install PyTorch with CUDA support on my Windows 11 machine, which has CUDA 12 installed and python 3. g. For further information on the compatible versions, check GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision for the compatibility matrix. device () API is a convenient wrapper for sending your tensors to the DirectML device. distributed. exe is. ) Enter these commands, which will install webui to your current directory: sudo apt install git python3. py", line 167, in exc_logging_wrapper. The Stable-Diffusion-v1-5 checkpoint was initialized with the weights of the Stable-Diffusion-v1-2 checkpoint and subsequently fine-tuned on 595k steps at resolution 512x512 on "laion-aesthetics v2 5+" and 10% dropping of the text-conditioning to improve classifier-free guidance sampling. PyTorch with DirectML provides an easy-to-use way for developers to try out the latest and greatest AI models on their Windows machine. Run this command Run the command `pip install “path to the downloaded WHL file” –force-reinstall` to install the package. (If you use this option, make sure to select “ Add Python to 3. Microsoft has provided a path in DirectML for vendors like AMD to enable optimizations called ‘metacommands’. You can download PyTorch with DirectML by installing the torch-directml PyPi package. You may remember from this year’s Build that we showcased Olive support for Stable Diffusion, a cutting-edge Generative AI model that creates images from text. If you have another Stable Diffusion UI you might be able to reuse the DirectML, a powerful machine learning API developed by Microsoft, is fast, versatile, and works seamlessly across a wide range of hardware platforms. With the release of the latest Intel® Arc™ GPU, we’ve gotten quite a few questions about whether the Intel Arc card supports running Tensorflow and PyTorch models, and the answer is YES! Built using the oneAPI specification, the Intel Download the stable release or the most recent SHARK 1. This project is aimed at becoming SD WebUI's Forge. This repository implements Stable Diffusion. rank_zero_only` has been deprecated in v1. 1 doesn't support PyTorch 2. Proceeding without it. Nov 9, 2022 · One of the easiest ways to try Stable Diffusion is through the Hugging Face Diffusers library. RunwayML Stable Diffusion 1. The issue exists in the current version of the webui. 5, 2. yaml C: \U sers \~\s table-diffusion-webui-directml \v env \l ib \s ite-packages \h uggingface_hub \f ile_download. It won't work on Windows 10 It won't work on Windows 10 If there is a better perf on Linux drivers, you won't be getting them with the above method. Run save_onnx. If you have custom models put them in a models/ directory where the . If you have another Stable Diffusion UI you might be able to reuse the Stable Diffusion 3 represents a major leap forward in the capability of AI to generate bespoke and high-fidelity images from text prompts. (and there's no available distribution of torch-directml for Linux) Or you can try with ROCm. In the case of Stable Diffusion with the Olive pipeline, AMD has released driver support for a metacommand implementation intended to Oct 21, 2021 · October 21st, 2021 3 0. Olive is an easy-to-use hardware-aware model optimization tool that composes industry-leading techniques across model compression, optimization, and compilation. Nov 14, 2023 · Click to share on Twitter (Opens in new window) Click to share on Facebook (Opens in new window) March 24, 2023. Install the ComfyUI dependencies. Double click the . 10, now It seems to properly download pytorch. py --help. Here, we will use Stable Diffusion XL 1. For instructions, read the Accelerated PyTorch training on Mac Apple Developer guide (make sure to install the latest pytorch nightly). There is a known issue I've been researching, and I think it boils down to the user needing to execute the script webui. I tried fp16 instead but that just breaks elsewhere. AMD RX5700XT GPU with AMD 3900x CPU and 64gb 3200mhz RAM Automatic1111 Stable Diffusion v1. The extension uses ONNX Runtime and DirectML to run inference against these models. I've enabled the ONNX runtime in settings, enabled Olive in settings (along with all the check boxes required) added the sd_unet checkpoint model thing (whatever you call it) under The optimized model will be stored at the following directory, keep this open for later: olive\examples\directml\stable_diffusion\models\optimized\runwayml. In launch. 5; Once you have selected a model version repo, click Files and Versions, then select the ONNX branch. After following some tutorials to install directml (i basically just created a conda venv and installed Pytorch-directml after some plugins) and the code in his video that he uses to time gpu and cpu take me respectively, for 5000 particles, 6. In these cases, users will have to manually add the models themselves. If you have another Stable Diffusion UI you might be able to reuse the Sep 4, 2022 · こんにちは、 @kz_morita です。 今回は、話題の Stable Diffusion を WSL 上の Ubuntu で動かしてみました。ところどころ詰まったところがあったのでメモを残していきます。 環境について 筆者のPC環境は以下の通りです。 Windows 10 Ubuntu 20. Apr 2, 2023 · Execute the webui. Processing without No module 'xformers'. 7, and 3. sh {your_arguments*} *For many AMD GPUs, you must add --precision full --no-half or --upcast-sampling arguments to avoid NaN errors or crashing. None of these seem to make a difference. 0 at this time. Make sure to set the MODEL_NAME and DATASET_NAME environment variables. Contribute to leejet/stable-diffusion. models. Mar 17, 2024 · D:\GitResource\stable-diffusion-webui-directml\venv\lib\site-packages\pytorch_lightning\utilities\distributed. py --interactive --num_images 2. Fooocus is an image generating software (based on Gradio ). ddpm import LatentDiffusion File "G:\OLIVESD\stable-diffusion-webui-directml\repositories\stable-diffusion-stability-ai\ldm\models\diffusion\ddpm. Feb 16, 2024 · D:\AI\Applications\Stable Diffusion (ZLUDA)\stable-diffusion-webui-directml\venv\lib\site-packages\pytorch_lightning\utilities\distributed. First start an interactive Python session, and import Torch with the following lines: Copy. sudo apt install wget git python3 python3-venv libgl1 libglib2. py –help. The issue has not been reported before recently. 0-0. Stable Diffusion Models v1. cpp development by creating an account on GitHub. - microsoft/Olive March 24, 2023. exe, or run from the command line (recommended), and you should have the UI in the browser. com/AUTOMATIC1111/stable-diffusion-webui && cd stable-diffusion-webui. I've had plenty of issues with my setup but it all boiled down to: remove all SD, pyton/pytorch, etc. You must have Windows or WSL environment to run DirectML. March 24, 2023. Please ensure that you have met the May 17, 2023 · 9). bat venv "C:\Users\alias\stable-diffusion-webui-directml\venv\Scripts\Python. 12]: Added more new features in WebUI extension, see the discussion here. In conclusion, DirectML can't use PyTorch 2. 8, and includes support for GPU device selection. UPDATE: A faster (20x) approach for running Stable Diffusion using MLIR/Vulkan/IREE is available on Windows: https://github. Embedded Git and Python dependencies, with no need for either to be globally installed. As of today the repo provides code to do the following: Training and Inference on Unconditional Latent Diffusion Models; Training a Class Conditional Latent Diffusion Model; Training a Text Conditioned Latent Diffusion Model; Training a Semantic Mask Conditioned Latent Diffusion Model Apr 22, 2023 · When I run webui-user. Aug 9, 2023 · DirectML depends on DirectX api. This was mainly intended for use with AMD GPUs but should work just as well with other DirectML devices (e. 05]: Released a new 512x512px (beta) face model. rank_zero_only has been deprecated in v1. Dec 8, 2022 · I experimented a bit but it doesn't offer a lot of pointers as to what is really wrong. 5 HFID runwayml/stable-diffusion-v1-5 L Text-to-Image with Stable Diffusion. Sep 22, 2022 · 4. sudo apt update && sudo apt upgrade. This model allows for image variations and mixing operations as described in Hierarchical Text-Conditional Image Generation with CLIP Latents, and, thanks to its modularity, can be combined with other models such as KARLO. 5; Stable Cascade Full and Lite; aMUSEd 256 256 and 512; Segmind Vega; Segmind Install pytorch nightly. We take care of UI/UX design, witch in my opinion better then NMKD. md. 1; LCM: Latent Consistency Models; Playground v1, v2 256, v2 512, v2 1024 and latest v2. Dec 24, 2023 · Please add --use-directml to skip CUDA test. sh in the root folder (execute with bash or similar) and it should install ROCM. DirectML in action. Automatic Installation. The name "Forge" is inspired from "Minecraft Forge". 👍 6 Gozhack, jchulce, eggressive, Gloryandel, VekuDazo, and erenustun reacted with thumbs up emoji Feb 17, 2024 · ModuleNotFoundError: No module named 'keras. 6 Checklist The issue exists after disabling all extensions The issue exists on a clean installation of webui The issue is caused by an extension, but I believe it is caused by a bug in the webui The issue exists in the current version of Features: settings tab rework: add search field, add categories, split UI settings page into many. See the Apr 10, 2023 · Saved searches Use saved searches to filter your results more quickly Feb 8, 2024 · no module 'xformers'. py:258: LightningDeprecationWarning: `pytorch_lightning. x). py", line 384, in prepare_environment raise RuntimeError( RuntimeError: Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check Apr 16, 2024 · And the model folder will be named as: “stable-diffusion-v1-5” If you want to check what different models are supported then you can do so by typing this command: python stable_diffusion. zluda. When I run nvcc --version, I get the following output: nvcc: NVIDIA (R) Cuda The previous changelog can be found here. deleted the folders stable-diffusion-webui\models\Stable-diffusion, LORA, VAE, VAE-approx Opened cmd. 5 on Ubuntu 22. 4. For Stable Diffusion in Linux/WSL using IPEX (Intel Extensions for Pytorch) see here. py", line 560, in prepare_environment raise RuntimeError( RuntimeError: Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check May 21, 2024 · ONNX Runtime with DirectML applies state-of-the-art optimizations to get the best performance for all generative AI models like Phi, Llama, Mistral, and Stable Diffusion. Non-converted Pytorch Models Stable Diffusion 1. We provide a reference script for sampling, but there also exists a diffusers integration, which we expect to see more active community development. Customization: Create your own presets - select a Stable Diffusion checkpoint, add LoRA, tweak samplers and more. device() The current release of torch-directml is mapped to the "PrivateUse1" Torch backend. Resources for more information: SDXL paper on arXiv. The model folder will be called “stable-diffusion-v1-5”. Jun 6, 2024 · Stable Diffusion for AMD GPUs on Windows using DirectML. On the DirectML GitHub, you’ll find a new public Operator Roadmap indicating current and planned Manage plugins / extensions for supported packages ( Automatic1111, Comfy UI, SD Web UI-UX, and SD. In some regions, UL Procyon cannot automatically download the required AI models. 1 which means it is for PyTorch 1. utilities. com/nod-ai/SHARK/blob/main/shark/examples/shark_inference/stable_diffusion/stable_diffusion_amd. The DirectML backend for Pytorch enables high-performance, low-level access to the GPU hardware, while exposing a familiar Pytorch API for developers. 10. 1 are supported. Intel Arc). The issue exists after disabling all extensions. py", line 1086, in get_module. . With support from every DirectX 12-capable GPU and soon across NPUs, developers can use DirectML to deliver AI experiences at scale. An installable Python package is now hosted on pytorch. 6, 3. x (all variants) StabilityAI Stable Diffusion XL; StabilityAI Stable Diffusion 3 Medium; StabilityAI Stable Video Diffusion Base, XT 1. sh and pytorch+rocm should be automatically installed for you. 1 and will be removed in v2. You can reuse your favorite Python packages such as NumPy, SciPy, and Cython to extend PyTorch when needed. Preview is available if you want the latest, not fully tested and supported, builds that are generated nightly. Youshould be able to run pytorch with directml inside wsl2, as long as you have latest AMD windows drivers and Windows 11. Next) Easily install or update Python dependencies for each package. 04 on WSL2 NVIDIA GeForce RTX 3060Ti (VRAM 8GB) 環境構築 Windows のバージョン Mar 26, 2023 · Doing changes with already present SD installation often break things. 8 release, we are delighted to announce a new installation option for users of PyTorch on the ROCm™ open software platform. Deep neural networks built on a tape-based autograd system. The torch_directml. Once set up, you can start with our samples or use the AI Toolkit for VS Code. x and 2. Dec 24, 2023 · File "C:\AI\stable-diffusion-webui\modules\launch_utils. Use the following command to see what other models are supported: python stable_diffusion. But Linux systems do not have it. Hello everyone. 0 and 2. Using ZLUDA in C:\Users\alias\stable-diffusion-webui-directml. With the PyTorch 1. 1. With its enhanced technical framework and strong performance against competitors like Midjourney and Dall-E 3, SD3 is poised to become a leading tool in the creative industries, offering users unprecedented control and quality in visual content creation. py:258: LightningDeprecationWarning: pytorch_lightning. import torch_directml. I tried my best to make the codebase minimal, self-contained, consistent, hackable, and easy to read. distributed' Your filepath should look something like this: "C:\Users\NAME\Stable-diffusion-webui\stable-diffusion-webui\venv In the File Explorer bar at the top type 'cmd' to open the standard command prompt inside the venv directory Stable Diffusion XUI for Nvidia and AMD GPU. UPDATE: Nearly all AMD GPU's from the RX470 and above are now working. This preview extension offers DirectML support for compute-heavy uNet models in Stable Diffusion, similar to Automatic1111's sample TensorRT extension and NVIDIA's TensorRT extension. 95~3 IT/s on a RX 5700 XT. 04. exe and created symbolic links to my old installation that has all of my checkpoints, LORAs, etc example: c: May 21, 2024 · We built some samples to show how you can use DirectML and the ONNX Runtime: Phi-3-mini; Large Language Models (LLMs) Stable Diffusion; Style transfer; Inference on NPUs; DirectML and PyTorch. Features are pruned if not needed in Stable Diffusion (e. distributed import rank_zero_only ModuleNotFoundError: No module named 'pytorch_lightning. ij nv vi ix mw qy jr ul ak wa