From ae9fae92c347b712065ba8987b51fdae64f87fea Mon Sep 17 00:00:00 2001
From: robertgro <34941707+robertgro@users.noreply.github.com>
Date: Thu, 28 Nov 2024 17:09:57 +0100
Subject: [PATCH] Update tiny_txt2img.py

required for CUDA to work, see https://github.com/Xilinx/Vitis-AI/issues/822#issuecomment-1151970403
---
 tiny_optimizedSD/tiny_txt2img.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tiny_optimizedSD/tiny_txt2img.py b/tiny_optimizedSD/tiny_txt2img.py
index e09b7d5..870a204 100644
--- a/tiny_optimizedSD/tiny_txt2img.py
+++ b/tiny_optimizedSD/tiny_txt2img.py
@@ -1,5 +1,5 @@
 import argparse, os, re, sys
-os.environ['CUDA_VISIBLE_DEVICES'] = '2'
+os.environ['CUDA_VISIBLE_DEVICES'] = '0'
 sys.path.append('.')
 sys.path.append('/home/hyf/latent-diffusion/taming-transformers')
 import torch