From 9867b4c85d4ff242be38fadbc103f64796b479ff Mon Sep 17 00:00:00 2001
From: Martin Evans <martindevans@gmail.com>
Date: Sat, 27 Apr 2024 02:55:35 +0100
Subject: [PATCH] Only setting callback if the token can be cancelled.

---
 LLama/LLamaWeights.cs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/LLama/LLamaWeights.cs b/LLama/LLamaWeights.cs
index ad041aac..e37de1e9 100644
--- a/LLama/LLamaWeights.cs
+++ b/LLama/LLamaWeights.cs
@@ -107,7 +107,7 @@ namespace LLama
             {
 #if !NETSTANDARD2_0
                 // Overwrite the progress callback with one which polls the cancellation token
-                //if (token.CanBeCanceled)
+                if (token.CanBeCanceled)
                 {
                     var internalCallback = lparams.progress_callback;
                     lparams.progress_callback = (progress, ctx) =>