diff --git a/requirements.txt b/requirements.txt index ed33d32..46e0398 100644 --- a/requirements.txt +++ b/requirements.txt @@ -88,6 +88,7 @@ test-tube>=0.7.5 timm==0.6.7 torch-fidelity==0.3.0 transformers==4.19.2 # do not change +torchmetrics==0.11.4 wget # Optional packages commonly used with Stable Diffusion workflow