diff --git a/README.md b/README.md
index 0229d0b..78c6bd3 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,5 @@
 # ComfyUI ExLlama Nodes
-A simple prompt generator for [ComfyUI](https://github.com/comfyanonymous/ComfyUI) using [ExLlama](https://github.com/turboderp/exllama).
+A simple prompt generator for [ComfyUI](https://github.com/comfyanonymous/ComfyUI) utilizing [ExLlama](https://github.com/turboderp/exllama).
 
 ## Installation
 Clone the repository to `custom_nodes` in your ComfyUI directory and install dependencies:
diff --git a/nodes.py b/nodes.py
index 6846e12..de97c1b 100644
--- a/nodes.py
+++ b/nodes.py
@@ -6,9 +6,9 @@ from colorama import Fore
 from comfy.utils import ProgressBar
 
 if not torch.cuda.is_available():
-    raise Exception(f"\n{Fore.RED}No CUDA detected. ExLlama doesn't support CPU.{Fore.RESET}")
+    raise Exception(f"\n{Fore.RED}ExLlama doesn't support CPU mode.{Fore.RESET}")
 
-cuda = torch.version.cuda.replace(".", "")
+cuda = str(torch.version.cuda or torch.version.hip).replace(".", "")
 pckg = f"cu{cuda}-cp{sys.version_info.major}{sys.version_info.minor}"
 
 try:
diff --git a/scripts.js b/scripts.js
index 7d7b2bb..d4afeda 100644
--- a/scripts.js
+++ b/scripts.js
@@ -19,7 +19,6 @@ app.registerExtension({
 						for (let i = position; i < this.widgets.length; i++) {
 							this.widgets[i].onRemove?.();
 						}
-
 						this.widgets.length = position;
 					}