more attention types

This commit is contained in:
layerdiffusion
2024-08-02 03:32:19 -07:00
parent 318219bc9d
commit 3dd084d55b
5 changed files with 699 additions and 45 deletions

View File

@@ -1,6 +1,6 @@
import torch
import os
import safetensors
import safetensors.torch
def build_loaded(module, loader_name):