Fix config

basetenlabs · Nov 8, 2023 · ecc34ba · ecc34ba
1 parent 1914398
commit ecc34ba
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/gfp-gan/config.yaml b/gfp-gan/config.yaml
@@ -19,4 +19,9 @@ system_packages:
 - ffmpeg
 - libsm6
 - libxext6
-spec_version: 2.0
+spec_version: 2.0
+external_data:
+  - url: https://baseten-public.s3.us-west-2.amazonaws.com/models/gfp-gan/RealESRGAN_x2plus.pth
+    local_data_path: RealESRGAN_x2plus.pth
+  - url: https://baseten-public.s3.us-west-2.amazonaws.com/models/gfp-gan/GFPGANv1.3.pth
+    local_data_path: GFPGANv1.3.pth