Merge branch 'master' into master

This commit is contained in:
brkirch
2022-10-04 01:04:19 -04:00
committed by GitHub
40 changed files with 1610 additions and 426 deletions

View File

@@ -32,10 +32,9 @@ def enable_tf32():
errors.run(enable_tf32, "Enabling TF32")
device = get_optimal_device()
device_gfpgan = device_codeformer = cpu if device.type == 'mps' else device
dtype = torch.float16
def randn(seed, shape):
# Pytorch currently doesn't handle setting randomness correctly when the metal backend is used.