From 5ea49e6af249e6eabca5ee89de76182975842ec2 Mon Sep 17 00:00:00 2001
From: tjandy98 <3953059+tjandy98@users.noreply.github.com>
Date: Thu, 7 May 2026 14:49:17 +0800
Subject: [PATCH 1/2] Update worker.go

Signed-off-by: tjandy98 <3953059+tjandy98@users.noreply.github.com>
---
 pkg/service/worker.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pkg/service/worker.go b/pkg/service/worker.go
index 8628736..98dd49f 100644
--- a/pkg/service/worker.go
+++ b/pkg/service/worker.go
@@ -165,8 +165,10 @@ func (worker *Worker) pullModel(ctx context.Context, statusPath, volumeName, mou
 		}
 		defer worker.kmutex.Unlock(contextKey)
 
+		// Decouple from the kubelet RPC deadline so that large pulls aren't killed
+		// when kubelet times out and retries
 		var cancel context.CancelFunc
-		ctx, cancel = context.WithCancel(ctx)
+		ctx, cancel = context.WithCancel(context.WithoutCancel(ctx))
 		worker.contextMap.Set(contextKey, &cancel)
 		defer worker.contextMap.Set(contextKey, nil)
 

From b399b63932613f1bcd77906ea09bf4eb4a7701ac Mon Sep 17 00:00:00 2001
From: tjandy98 <3953059+tjandy98@users.noreply.github.com>
Date: Thu, 7 May 2026 16:48:06 +0800
Subject: [PATCH 2/2] Update node_static_inline.go

Signed-off-by: tjandy98 <3953059+tjandy98@users.noreply.github.com>
---
 pkg/service/node_static_inline.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pkg/service/node_static_inline.go b/pkg/service/node_static_inline.go
index cc9add1..4e4f9dc 100644
--- a/pkg/service/node_static_inline.go
+++ b/pkg/service/node_static_inline.go
@@ -25,8 +25,10 @@ func (s *Service) nodePublishVolumeStaticInlineVolume(ctx context.Context, volum
 	duration := time.Since(startedAt)
 	logger.WithContext(ctx).Infof("pulled model: %s %s", reference, duration)
 
+	mountCtx, mountCancel := context.WithTimeout(context.WithoutCancel(ctx), 30*time.Second)
+	defer mountCancel()
 	if err := mounter.Mount(
-		ctx,
+		mountCtx,
 		mounter.NewBuilder().
 			Bind().
 			From(modelDir).