skytnt
/

deepdanbooru_onnx

Model card Files Files and versions Community

skytnt commited on Jan 30, 2023

Commit

817c2b7

•

1 Parent(s): 23d2ee2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -140,7 +140,7 @@ batch_size = 8
 tagger_model_path = hf_hub_download(repo_id="skytnt/deepdanbooru_onnx", filename="deepdanbooru.onnx")
 tagger_model = [
     rt.InferenceSession(tagger_model_path, providers=['CUDAExecutionProvider'], provider_options=[{'device_id': i}]) for
-    i in range(4)]
 tagger_model_meta = tagger_model[0].get_modelmeta().custom_metadata_map
 tagger_tags = eval(tagger_model_meta['tags'])

 tagger_model_path = hf_hub_download(repo_id="skytnt/deepdanbooru_onnx", filename="deepdanbooru.onnx")
 tagger_model = [
     rt.InferenceSession(tagger_model_path, providers=['CUDAExecutionProvider'], provider_options=[{'device_id': i}]) for
+    i in range(gpu_num)]
 tagger_model_meta = tagger_model[0].get_modelmeta().custom_metadata_map
 tagger_tags = eval(tagger_model_meta['tags'])