fix(finetune): use the old restart for finetune sessions

helixml · Aug 28, 2024 · 6f74039 · 6f74039
1 parent c1ec717
commit 6f74039
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/pkg/server/handlers.go b/api/pkg/server/handlers.go
@@ -648,7 +648,7 @@ func (apiServer *HelixAPIServer) restartSession(res http.ResponseWriter, req *ht
 		return nil, err
 	}
 	// If it is a text inference session, then restart using the "new" openai controllers
-	if session.Type == types.SessionTypeText && session.Mode == types.SessionModeInference {
+	if session.Metadata.OriginalMode != types.SessionModeFinetune && session.Type == types.SessionTypeText && session.Mode == types.SessionModeInference {
 		apiServer.restartChatSessionHandler(res, req)
 		return session, nil
 	}