style: cleaned up a few print statements.

coreweave · May 19, 2024 · 741198f · 741198f
1 parent 11c7b17
commit 741198f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/gpt_bpe_test.go b/gpt_bpe_test.go
@@ -812,7 +812,6 @@ func TestLlamaTwoEncodeDecode(t *testing.T) {
 func TestMistralEncoder_Encode(t *testing.T) {
 	testString := "The fox jumped over the hare.\nThe turtle is faster than the hare."
 	mistralTokens := mistralEncoder.Encode(&testString)
-	fmt.Printf("mistralTokens: %v\n", mistralTokens)
 	assert.Equal(t, mistralTokens, &Tokens{1, 415, 285, 1142, 14949, 754, 272, 295, 492, 28723, 13, 1014, 261, 3525, 291, 349, 9556, 821, 272, 295, 492, 28723})
 }
 

diff --git a/resources/resolver.go b/resources/resolver.go
@@ -84,7 +84,7 @@ func (rsrc *Resources) GetFile(name string) (interface{}, error) {
 	if rsrcEntry, ok := (*rsrc)[name]; ok {
 		return rsrcEntry.file, nil
 	} else {
-		return nil, errors.New(fmt.Sprintf("resource %s not found", name))
+		return nil, fmt.Errorf("file %s not found", name)
 	}
 }
 
@@ -811,7 +811,7 @@ func ResolveHFFromResources(resources *Resources, hfConfig *HFConfig) (*HFConfig
 			return nil, err
 		}
 	} else {
-		fmt.Printf("config.json not found in resources\n error: %s", err)
+		log.Printf("Model file %s, will attempt to skip\n", err)
 	}
 
 	if _, err := resources.GetFile("tokenizer_config.json"); err == nil {
@@ -820,7 +820,7 @@ func ResolveHFFromResources(resources *Resources, hfConfig *HFConfig) (*HFConfig
 			return nil, err
 		}
 	} else {
-		fmt.Printf("tokenizer_config.json not found in resources\n error: %s", err)
+		log.Printf("Model file: %s, will attempt to skip\n", err)
 
 	}
 
@@ -886,7 +886,6 @@ func ResolveHFFromResources(resources *Resources, hfConfig *HFConfig) (*HFConfig
 		}
 
 	}
-	fmt.Printf("Resolved config: %v\n", &hfConfig)
 	return hfConfig, nil
 }
 
@@ -895,7 +894,7 @@ func ResolveHFFromResources(resources *Resources, hfConfig *HFConfig) (*HFConfig
 // local filesystem, or remote.
 func ResolveVocabId(vocabId string, token string) (*HFConfig, *Resources, error) {
 	var resolvedVocabId string
-	fmt.Printf("Resolving vocab id: %s\n", vocabId)
+	log.Printf("Resolving vocab id: %s\n", vocabId)
 	if _, vocabErr := EmbeddedDirExists(vocabId); vocabErr == nil {
 		endOfText := "<|endoftext|>"
 		bosText := "<|startoftext|>"