HiDolen
/

Mini-BS-RoFormer-V2-46.8M

feature-extraction

Model card Files Files and versions

HiDolen commited on Dec 30, 2025

Commit

2907229

·

verified ·

1 Parent(s): 8b43f9d

Update modeling_bs_roformer.py

Files changed (1) hide show

modeling_bs_roformer.py +1 -1

modeling_bs_roformer.py CHANGED Viewed

@@ -507,7 +507,7 @@ class BSRoformerForMaskedEstimation(BSRoformerPreTrainedModel):
         mixed_wave,
         chunk_size=None,
         overlap_size=None,
-        batch_size=4,
         gap_size=44100 * 0,
         verbose=True,
     ):

         mixed_wave,
         chunk_size=None,
         overlap_size=None,
+        batch_size=1,
         gap_size=44100 * 0,
         verbose=True,
     ):