Skip to content

Commit 462a196

Browse files
committed
fix infinite hang during GFPGAN duration inadvertently introduced during batch_size cleanup
2 parents c83d01b + 2aa8393 commit 462a196

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

environment-mac.yaml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,3 +30,5 @@ dependencies:
3030
- -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
3131
- -e git+https://github.com/lstein/k-diffusion.git@master#egg=k-diffusion
3232
- -e .
33+
variables:
34+
PYTORCH_ENABLE_MPS_FALLBACK: 1

ldm/simplet2i.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def __init__(
124124
grid=False,
125125
width=512,
126126
height=512,
127-
sampler_name='klms',
127+
sampler_name='k_lms',
128128
latent_channels=4,
129129
downsampling_factor=8,
130130
ddim_eta=0.0, # deterministic

0 commit comments

Comments
 (0)