Fix version string with rocm

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:

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:

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;
 					}