georgefen's picture
add all needed files
1cb032f
raw
history blame
155 Bytes
import config
from cldm.hack import disable_verbosity, enable_sliced_attention
disable_verbosity()
if config.save_memory:
enable_sliced_attention()