From c76226571e69ffad52386f57b65fd3ef2d051051 Mon Sep 17 00:00:00 2001
From: Brett Kuprel <brkuprel@gmail.com>
Date: Wed, 29 Jun 2022 19:14:25 -0400
Subject: [PATCH] Update cog.yaml

---
 cog.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cog.yaml b/cog.yaml
index 4e0f631..42e8f9b 100644
--- a/cog.yaml
+++ b/cog.yaml
@@ -7,6 +7,6 @@ build:
     - "libglib2.0-0"
   python_packages:
     - "torch==1.10.1"
-    - "flax==0.4.2"
+    - "flax==0.5.2"
 
 predict: "predict.py:Predictor"