Commit
·
07e3040
1
Parent(s):
09790b2
update
Browse files
nested_attention_pipeline.py
CHANGED
|
@@ -4,7 +4,7 @@ from typing import List
|
|
| 4 |
import torch
|
| 5 |
from PIL import Image
|
| 6 |
from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
|
| 7 |
-
from safetensors import load_file
|
| 8 |
|
| 9 |
from nested_attention_processor import AttnProcessor, NestedAttnProcessor
|
| 10 |
from utils import get_generator
|
|
|
|
| 4 |
import torch
|
| 5 |
from PIL import Image
|
| 6 |
from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
|
| 7 |
+
from safetensors.torch import load_file
|
| 8 |
|
| 9 |
from nested_attention_processor import AttnProcessor, NestedAttnProcessor
|
| 10 |
from utils import get_generator
|