ifckpt_path.endswith("_shuffle.pth")orckpt_path.endswith("_shuffle.safetensors")orckpt_path.endswith("_shuffle_fp16.safetensors"):#TODO: smarter way of enabling global_average_pooling
filename=os.path.splitext(ckpt_path)[0]
iffilename.endswith("_shuffle")orfilename.endswith("_shuffle_fp16"):#TODO: smarter way of enabling global_average_pooling