feat: 设置输出格式为jpg

codyttang · codyttang · commit e984a4dafe3c · 2025-05-07T17:53:58.000+08:00
diff --git a/configs/prediction/default.yaml b/configs/prediction/default.yaml
@@ -7,11 +7,12 @@ model:
 
 dataset:
   kind: default
-  img_suffix: .jpg
+  img_suffix: .png
   pad_out_to_modulo: 8
 
 device: cuda
 out_key: inpainted
+out_ext: .jpg
 
 refine: False # refiner will only run if this is True
 refiner:
