requirements.txt: Added TensorFlow Probability module

Signed-off-by: Grzegorz Latosinski <[email protected]>
antmicro · Apr 2, 2024 · a397cb6 · a397cb6
1 parent 97ea0ca
commit a397cb6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -8,6 +8,7 @@ Pillow
 scikit_learn
 tensorflow==2.13.0
 tensorflow_model_optimization
+tensorflow_probability==0.21.0
 tflite
 torch
 torchvision