Fix a bug fetching dataset from HF

Signed-off-by: Sheng Lin <[email protected]>

Author: shengnuo

api/apps/v1alpha1/nimcache_types.go

@@ -211,7 +211,7 @@ const (
 
 // EnvFromSecrets return the list of secrets that should be mounted as env vars.
 func (s *NIMSource) EnvFromSecrets() []corev1.EnvFromSource {
-	if s.NGC != nil && s.NGC.AuthSecret != "" {
+	if s.NGC != nil && s.NGC.AuthSecret != "" { // nolint:gocritic
 		return []corev1.EnvFromSource{
 			{
 				SecretRef: &corev1.SecretEnvSource{
@@ -231,6 +231,16 @@ func (s *NIMSource) EnvFromSecrets() []corev1.EnvFromSource {
 				},
 			},
 		}
+	} else if s.HF != nil && s.HF.AuthSecret != "" {
+		return []corev1.EnvFromSource{
+			{
+				SecretRef: &corev1.SecretEnvSource{
+					LocalObjectReference: corev1.LocalObjectReference{
+						Name: s.HF.AuthSecret,
+					},
+				},
+			},
+		}
 	}
 	// no secrets to source the env variables
 	return []corev1.EnvFromSource{}

internal/controller/nimcache_controller.go

@@ -1137,7 +1137,7 @@ func (r *NIMCacheReconciler) constructJob(ctx context.Context, nimCache *appsv1a
 		job.Spec.Template.Spec.Containers = []corev1.Container{
 			{
 				Name:    NIMCacheContainerName,
-				Image:   nimCache.Spec.Source.DataStore.ModelPuller,
+				Image:   hfDataSource.GetModelPuller(),
 				EnvFrom: nimCache.Spec.Source.EnvFromSecrets(),
 				Env: []corev1.EnvVar{
 					{