From 270722cd3a7aeb0d83095286f2f09e80fdd137c0 Mon Sep 17 00:00:00 2001 From: Markus F <39213730+markus583@users.noreply.github.com> Date: Fri, 28 Jun 2024 17:00:40 +0200 Subject: [PATCH] Update README.md --- README.md | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index c1fdba65..76891531 100644 --- a/README.md +++ b/README.md @@ -23,14 +23,14 @@ from wtpsplit import SaT sat = SaT("sat-3l") # optionally run on GPU for better performance -# also supports TPUs via e.g. wtp.to("xla:0"), in that case pass `pad_last_batch=True` to wtp.split +# also supports TPUs via e.g. sat.to("xla:0"), in that case pass `pad_last_batch=True` to sat.split sat.half().to("cuda") # returns ["This is a test", "This is another test."] sat.split("This is a test This is another test.") # returns an iterator yielding a lists of sentences for every text -# do this instead of calling wtp.split on every text individually for much better performance +# do this instead of calling sat.split on every text individually for much better performance sat.split(["This is a test This is another test.", "And some more texts..."]) # use our '-sm' models for general sentence segmentation tasks @@ -40,7 +40,8 @@ sat.split("this is a test this is another test") # returns ["this is a test", "this is another test"] # use trained lora modules for strong adaptation to language & domain/style -sat.split("This is a test This is another test.", lang_code="en", style="ud") +sat_adapted = SaT("sat-3l-sm", lang_code="en", style="ud")) +sat.split("This is a test This is another test.") ```