hadriansecurity / hadriansecurity/subwiz

Implement batch inference for multi-apex

Open
#25 6 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
391
Forks
25
PR merge metrics
No merged PRs in 30d

Description

Use batch inference for efficiency

Edit:

> when we perform inference for multi-apex we essentially do a for loop
> ```python
> gpt_model, tokenizer = get_model_and_tokenizer(force_download, device=device)
> found_domains = set()
>
> for apex in sorted(domain_groups):
> found_domains |= _get_domains_for_group(...)
>
> return sorted(found_domains)
> ```
>
> Here we are not making use of batch inference, as in we don't batch different apex related subdomains simultaneously.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.