Skip to content
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.

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

params_config for CPU and MPS #2436

Open
jie-chen opened this issue Jan 8, 2025 · 0 comments
Open

params_config for CPU and MPS #2436

jie-chen opened this issue Jan 8, 2025 · 0 comments

Comments

@jie-chen
Copy link

jie-chen commented Jan 8, 2025

For now, there are only three options to cater for different hardwares. Will there be a dynamic version or could anyone share some insight how to configure on MPS and CPU only machine?

if self.gpu_mem is not None and self.gpu_mem <= 4: x_pad = 1 x_query = 5 x_center = 30 x_max = 32 elif self.is_half: # 6G PU_RAM conf x_pad = 3 x_query = 10 x_center = 60 x_max = 65 else: # 5G GPU_RAM conf x_pad = 1 x_query = 6 x_center = 38 x_max = 41

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant