diff --git a/Project.toml b/Project.toml
index 6fd1b9f..da20b2f 100644
--- a/Project.toml
+++ b/Project.toml
@@ -57,7 +57,7 @@ TransformersCUDAExt = "CUDA"
 TransformersMetalExt = "Metal"
 
 [compat]
-AMDGPU = "0.9"
+AMDGPU = "0.9, 2"
 BangBang = "0.3, 0.4"
 BytePairEncoding = "0.5.1"
 CUDA = "5"