"examples/experimental/train_latent_text_to_image.py" did not exist on "bb3066428537da6263676448e737f315203d986c"
-
mshoeybi authored
Allocate tensor in `communicate()` method directly on GPU (instead of allocating on CPU and then moving to GPU)
9ff6f473