🌐 AI搜索 & 代理 主页
Skip to content

Allow user to limit the number of threads that OpenMP spawns. #1362

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
kczimm merged 15 commits into from
Apr 11, 2024
Merged

Allow user to limit the number of threads that OpenMP spawns. #1362

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
4612b4f
Refactor the initialization of GUC parameters.
higuoxing Mar 5, 2024
58e2aec
Allow user to limit the number of threads that OpenMP spawns.
higuoxing Mar 8, 2024
99bebe2
Merge branch 'master' into set-omp-num-threads
higuoxing Mar 15, 2024
58f8a9a
Merge branch 'master' into set-omp-num-threads
higuoxing Mar 23, 2024
2dc1964
Merge branch 'master' into set-omp-num-threads
higuoxing Apr 1, 2024
1361bce
Merge branch 'master' into set-omp-num-threads
higuoxing Apr 4, 2024
4e84e08
Address review comments.
higuoxing Apr 5, 2024
2a1853c
Address review comments
higuoxing Apr 5, 2024
3768179
Simplify global variables.
higuoxing Apr 6, 2024
6e3fe80
Fixup!!
higuoxing Apr 6, 2024
2e1891c
Merge branch 'master' into set-omp-num-threads
higuoxing Apr 10, 2024
d05644a
Update pgml-extension/src/bindings/transformers/whitelist.rs
higuoxing Apr 10, 2024
663ed25
Update pgml-extension/src/bindings/transformers/whitelist.rs
higuoxing Apr 10, 2024
c81c42e
Address review comments.
higuoxing Apr 10, 2024
ede3d92
Merge branch 'master' into set-omp-num-threads
kczimm Apr 11, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update pgml-extension/src/bindings/transformers/whitelist.rs 
Co-authored-by: kczimm <4733573+kczimm@users.noreply.github.com>
  • Loading branch information
@higuoxing @kczimm
higuoxing and kczimm authored Apr 10, 2024
commit d05644aa43f8c4045dbe745676cca85a789f9d91
8 changes: 1 addition & 7 deletions pgml-extension/src/bindings/transformers/whitelist.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,13 +33,7 @@ pub fn verify_task(task: &Value) -> Result<(), Error> {

let remote_code_allowed = trust_remote_code && model_is_trusted;
if !remote_code_allowed && task_trust == Some(true) {
bail!(
"model {} is not trusted to run remote code. Consider setting {} = 'true' or adding {} to {}",
task_model,
"pgml.huggingface_trust_remote_code",
task_model,
"pgml.huggingface_trust_remote_code_whitelist",
);
bail!("model {task_model} is not trusted to run remote code. Consider setting pgml.huggingface_trust_remote_code = 'true' or adding {task_model} to pgml.huggingface_trust_remote_code_whitelist");
}

Ok(())
Expand Down