Fix the OAuth check logic (#398)

huggingface · Dec 15, 2023 · f914cda · f914cda
1 parent 4d45439
commit f914cda
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/autotrain/trainers/common.py b/src/autotrain/trainers/common.py
@@ -51,7 +51,7 @@ def pause_space(params, is_failure=False):
             msg = f"Your training run was successful! [Check out your trained model here](https://huggingface.co/{params.repo_id})"
             title = "Your training has finished successfully ✅"
 
-        if params.token.startswith("hf_"):
+        if not params.token.startswith("hf_oauth_"):
             api.create_discussion(
                 repo_id=os.environ["SPACE_ID"],
                 title=title,