Commit18be006

committed

move embedding tests into their own file

1 parent455861e commit18be006Copy full SHA for 18be006

File tree

+10

-2

lines changed

+10

-2

lines changed

Lines changed: 7 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,7 @@`
	`1`	`+\timingon`
	`2`	`+`
	`3`	`+SELECTpgml.embed('Alibaba-NLP/gte-base-en-v1.5','hi mom','{"trust_remote_code": true}');`
	`4`	`+SELECTpgml.embed('Alibaba-NLP/gte-base-en-v1.5','hi mom','{"device": "cuda", "trust_remote_code": true}');`
	`5`	`+SELECTpgml.embed('Alibaba-NLP/gte-base-en-v1.5','hi mom','{"device": "cpu", "trust_remote_code": true}');`
	`6`	`+SELECTpgml.embed('hkunlp/instructor-xl','hi mom','{"instruction": "Encode it with love"}');`
	`7`	`+SELECTpgml.embed('mixedbread-ai/mxbai-embed-large-v1','test','{"prompt": "test prompt: "}');`

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -66,7 +66,7 @@ SELECT * FROM pgml.train('Handwritten Digits', algorithm => 'xgboost', hyperpara`
`66`	`66`
`67`	`67`	`-- runtimes`
`68`	`68`	`SELECT*FROMpgml.train('Handwritten Digits', algorithm=>'linear', runtime=>'python');`
`69`		`-SELECT*FROMpgml.train('Handwritten Digits', algorithm=>'linear', runtime=>'rust');`
	`69`	`+--SELECT * FROM pgml.train('Handwritten Digits', algorithm => 'linear', runtime => 'rust');`
`70`	`70`
`71`	`71`	`--SELECT * FROM pgml.train('Handwritten Digits', algorithm => 'xgboost', runtime => 'python', hyperparams => '{"n_estimators": 10}'); -- too slow`
`72`	`72`	`SELECT*FROMpgml.train('Handwritten Digits', algorithm=>'xgboost', runtime=>'rust', hyperparams=>'{"n_estimators": 10}');`

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,7 @@ SELECT pgml.transform_stream(`
`18`	`18`	`}'::JSONB,`
`19`	`19`	`input=>'AI is going to',`
`20`	`20`	`args=>'{`
`21`		`- "max_new_tokens": 100`
	`21`	`+ "max_new_tokens": 100,`
`22`	`22`	`}'::JSONB`
`23`	`23`	`);`
`24`	`24`	`-- BitsAndBytes support`

Lines changed: 1 addition & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -31,5 +31,6 @@ SELECT pgml.load_dataset('wine');`
`31`	`31`	`\i examples/vectors.sql`
`32`	`32`	`\i examples/chunking.sql`
`33`	`33`	`\i examples/preprocessing.sql`
	`34`	`+\i examples/embedding.sql`
`34`	`35`	`-- transformers are generally too slow to run in the test suite`
`35`	`36`	`--\i examples/transformers.sql`

Comments

(0)