diff --git a/codes/requirements.txt b/codes/requirements.txt index 7bb339a4..fd6ec31b 100644 --- a/codes/requirements.txt +++ b/codes/requirements.txt @@ -11,3 +11,7 @@ munch tqdm scp tensorboard +pytorch_fid +kornia +linear_attention_transformer +vector_quantize_pytorch \ No newline at end of file