diff --git a/src/rawdog/utils.py b/src/rawdog/utils.py index af0eca8..587c99c 100644 --- a/src/rawdog/utils.py +++ b/src/rawdog/utils.py @@ -91,4 +91,4 @@ def render_prompt(self): def is_finetuned_model(model: str): - return "ft:" in model or "rawdog" in model or "abante" in model + return "rawdog" in model or "abante" in model