PyTorch 2.0 Release

1. diffusers

   1 273 28,632 9.9 Python

   🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch and FLAX.

   

   In `diffusers` implementations (like InvokeAI) its pretty easy: https://github.com/huggingface/diffusers/blob/42beaf1d23b5cc...

   But I also compile the VAE and some other modules, I will reply again later when I can look at my local code.

   For the Automatic1111 repo (and presumably other original Stability AI implementations), I just add `m.model

2. Judoscale

   judoscale.com featured

   Save 47% on cloud hosting with autoscaling that just works. Judoscale integrates with Django, FastAPI, Celery, and RQ to make autoscaling easy and reliable. Save big, and say goodbye to request timeouts and backed-up task queues.

   

3. stable-diffusion-webui

   2 2,814 151,238 8.5 Python

   Stable Diffusion web UI

   

   = torch.compile(m.model)` here: https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob...

   I tried changing the options in the config dict one by one, but TBH nothing seems to make a significant difference.

   I also havent messed with compiling LORA training yet, as I havent done it often as is sufficiently fast, but I'm sure it could be done.

4. Pytorch

   3 387 88,992 10.0 Python

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   

   This is the master tracking list for MPS operator support: https://github.com/pytorch/pytorch/issues/77764

5. detectron2

   4 49 31,792 7.2 Python

   Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

   

   I could fine-tune a Detectron2 model a few months ago using PyTorch and MPS backend [1]. I'd be interested if it's working yet.

   https://github.com/facebookresearch/detectron2/issues/4342

Suggest a related project