• brkirch's avatar
    Use other MPS optimization for large q.shape[0] * q.shape[1] · 35b1775b
    brkirch authored
    Check if q.shape[0] * q.shape[1] is 2**18 or larger and use the lower memory usage MPS optimization if it is. This should prevent most crashes that were occurring at certain resolutions (e.g. 1024x1024, 2048x512, 512x2048).
    
    Also included is a change to check slice_size and prevent it from being divisible by 4096 which also results in a crash. Otherwise a crash can occur at 1024x512 or 512x1024 resolution.
    35b1775b
sd_hijack_optimizations.py 10.6 KB