Update yijinaembed.py

yijinaembed.py

@@ -25,9 +25,9 @@ from langchain.chains.combine_documents.stuff  import create_stuff_documents_cha
 
 load_dotenv()
 
-# os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:180'
-# os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
-# os.environ['CUDA_CACHE_DISABLE'] = '1'
+os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:180'
+os.environ['CUDA_LAUNCH_BLOCKING'] = '1'
+os.environ['CUDA_CACHE_DISABLE'] = '1'
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")