Removed autotune_cache

This commit is contained in:
Bruno Fargnoli
2025-12-29 00:14:27 +01:00
parent e933ce06a4
commit a481741854
2 changed files with 2 additions and 48759 deletions
-48757
View File
File diff suppressed because it is too large Load Diff
+2 -2
View File
@@ -165,8 +165,8 @@ class Trellis2LoadModel:
def process(self, modelname, backend, device, low_vram, keep_models_loaded):
os.environ['OPENCV_IO_ENABLE_OPENEXR'] = '1'
os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True" # Can save GPU memory
os.environ["FLEX_GEMM_AUTOTUNE_CACHE_PATH"] = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'autotune_cache.json')
os.environ["FLEX_GEMM_AUTOTUNER_VERBOSE"] = '1'
#os.environ["FLEX_GEMM_AUTOTUNE_CACHE_PATH"] = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'autotune_cache.json')
#os.environ["FLEX_GEMM_AUTOTUNER_VERBOSE"] = '1'
os.environ['ATTN_BACKEND'] = backend
torch.backends.cudnn.benchmark = False