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