Code

Fixes to run on CPU and MPS#36

Code

Open

WojtekKowaluk wants to merge 3 commits intoai-forever:mainai-forever/Kandinsky-2:mainfrom WojtekKowaluk:cpu-mps-fixWojtekKowaluk/Kandinsky-2:cpu-mps-fixCopy head branch name to clipboard

+23-19

Commits

Commits on May 14, 2023

• Fixes to run on CPU and MPS

  

  Wojtek Kowaluk

  authored and

  Wojtek Kowaluk

  committed
  4b3d3d5

Commits on May 20, 2023

• fix _extract_into_tensor on mps device

  

  Wojtek Kowaluk

  authored and

  Wojtek Kowaluk

  committed
  ef70508

• allow samplers to run on devices other than cuda

  

  Wojtek Kowaluk

  authored and

  Wojtek Kowaluk

  committed
  8068dd5