diff --git a/sam_aclip_pixart_sigma/__init__.py b/alfie/__init__.py similarity index 100% rename from sam_aclip_pixart_sigma/__init__.py rename to alfie/__init__.py diff --git a/sam_aclip_pixart_sigma/attn_processor.py b/alfie/attn_processor.py similarity index 100% rename from sam_aclip_pixart_sigma/attn_processor.py rename to alfie/attn_processor.py diff --git a/sam_aclip_pixart_sigma/cross_heatmap.py b/alfie/cross_heatmap.py similarity index 100% rename from sam_aclip_pixart_sigma/cross_heatmap.py rename to alfie/cross_heatmap.py diff --git a/sam_aclip_pixart_sigma/generate.py b/alfie/generate.py similarity index 100% rename from sam_aclip_pixart_sigma/generate.py rename to alfie/generate.py diff --git a/sam_aclip_pixart_sigma/grabcut.py b/alfie/grabcut.py similarity index 100% rename from sam_aclip_pixart_sigma/grabcut.py rename to alfie/grabcut.py diff --git a/sam_aclip_pixart_sigma/pipeline_pixart_sigma.py b/alfie/pipeline_pixart_sigma.py similarity index 100% rename from sam_aclip_pixart_sigma/pipeline_pixart_sigma.py rename to alfie/pipeline_pixart_sigma.py diff --git a/sam_aclip_pixart_sigma/self_heatmap.py b/alfie/self_heatmap.py similarity index 100% rename from sam_aclip_pixart_sigma/self_heatmap.py rename to alfie/self_heatmap.py diff --git a/sam_aclip_pixart_sigma/transformer_2d.py b/alfie/transformer_2d.py similarity index 100% rename from sam_aclip_pixart_sigma/transformer_2d.py rename to alfie/transformer_2d.py diff --git a/sam_aclip_pixart_sigma/trimap.py b/alfie/trimap.py similarity index 100% rename from sam_aclip_pixart_sigma/trimap.py rename to alfie/trimap.py diff --git a/sam_aclip_pixart_sigma/utils.py b/alfie/utils.py similarity index 100% rename from sam_aclip_pixart_sigma/utils.py rename to alfie/utils.py diff --git a/generate_prompt.py b/generate_prompt.py index 1e5de4b..792a400 100644 --- a/generate_prompt.py +++ b/generate_prompt.py @@ -2,7 +2,7 @@ from settings import parse_setting import json -from sam_aclip_pixart_sigma.generate import get_pipe, base_arg_parser, parse_bool_args +from alfie.generate import get_pipe, base_arg_parser, parse_bool_args from transformers import VitMatteImageProcessor, VitMatteForImageMatting @@ -11,11 +11,11 @@ from accelerate.logging import get_logger from accelerate.utils import set_seed -from sam_aclip_pixart_sigma.grabcut import grabcut, save_rgba +from alfie.grabcut import grabcut, save_rgba import torch -from sam_aclip_pixart_sigma.trimap import compute_trimap -from sam_aclip_pixart_sigma.utils import normalize_masks +from alfie.trimap import compute_trimap +from alfie.utils import normalize_masks torch.backends.cuda.matmul.allow_tf32 = True