feat: optimize cuda for vapoursynth

TensoRaws · Dec 10, 2024 · 87f9fc1 · 87f9fc1
1 parent a1b86d4
commit 87f9fc1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/example/vapoursynth.py b/example/vapoursynth.py
@@ -11,7 +11,7 @@
 # --- sisr, use fp16 to inference (vs.RGBH)
 
 model: BaseModelInterface = AutoModel.from_pretrained(
-    pretrained_model_name=ConfigType.RealESRGAN_AnimeJaNai_HD_V3_Compact_2x
+    pretrained_model_name=ConfigType.RealESRGAN_AnimeJaNai_HD_V3_Compact_2x, tile=None
 )
 
 clip = core.bs.VideoSource(source="s.mp4")
@@ -25,6 +25,7 @@
 # model: BaseModelInterface = AutoModel.from_pretrained(
 #     pretrained_model_name=ConfigType.RealESRGAN_AnimeJaNai_HD_V3_Compact_2x,
 #     fp16=False,
+#     tile=None
 # )
 #
 # clip = core.bs.VideoSource(source="s.mp4")