fix: use constraint instead of affinity for GPU models

fix: use constraint instead of affinity for GPU models #2