Backchange

ForserX · Apr 26, 2023 · 0375e2f · 0375e2f
1 parent 685f098
commit 0375e2f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/data/repo/diffusion_scripts/sd_xbackend.py b/data/repo/diffusion_scripts/sd_xbackend.py
@@ -10,7 +10,7 @@
 	from onnxruntime import SessionOptions
 	from modules.onnx.lora import blend_loras, buffer_external_data_tensors
 	from modules.onnx.textual_inversion import blend_textual_inversions
-	import torch_directml
+	#import torch_directml
 
 	ONNX_MODEL = "model.onnx"
 except :
@@ -51,7 +51,7 @@ class Device:
 
 	def __init__(self, device: str, fp):
 		if device == "onnx":
-			self.device = torch_directml.device(torch_directml.default_device())
+			self.device = "onnx" #torch_directml.device(torch_directml.default_device())
 		else:
 			self.device = device
 

diff --git a/data/requirements_onnx.txt b/data/requirements_onnx.txt
@@ -25,7 +25,6 @@ transformers-onnx
 accelerate
 ftfy
 regex
-torch_directml
 
 # DirectML Fix
 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/