refactoring

aimagelab · Aug 8, 2024 · 1ab3eb0 · 1ab3eb0
1 parent 21c880d
commit 1ab3eb0
Show file tree

Hide file tree

Showing 11 changed files with 4 additions and 4 deletions.
diff --git a/sam_aclip_pixart_sigma/__init__.py → alfie/__init__.py b/sam_aclip_pixart_sigma/__init__.py → alfie/__init__.py
diff --git a/sam_aclip_pixart_sigma/attn_processor.py → alfie/attn_processor.py b/sam_aclip_pixart_sigma/attn_processor.py → alfie/attn_processor.py
diff --git a/sam_aclip_pixart_sigma/cross_heatmap.py → alfie/cross_heatmap.py b/sam_aclip_pixart_sigma/cross_heatmap.py → alfie/cross_heatmap.py
diff --git a/sam_aclip_pixart_sigma/generate.py → alfie/generate.py b/sam_aclip_pixart_sigma/generate.py → alfie/generate.py
diff --git a/sam_aclip_pixart_sigma/grabcut.py → alfie/grabcut.py b/sam_aclip_pixart_sigma/grabcut.py → alfie/grabcut.py
diff --git a/...lip_pixart_sigma/pipeline_pixart_sigma.py → alfie/pipeline_pixart_sigma.py b/...lip_pixart_sigma/pipeline_pixart_sigma.py → alfie/pipeline_pixart_sigma.py
diff --git a/sam_aclip_pixart_sigma/self_heatmap.py → alfie/self_heatmap.py b/sam_aclip_pixart_sigma/self_heatmap.py → alfie/self_heatmap.py
diff --git a/sam_aclip_pixart_sigma/transformer_2d.py → alfie/transformer_2d.py b/sam_aclip_pixart_sigma/transformer_2d.py → alfie/transformer_2d.py
diff --git a/sam_aclip_pixart_sigma/trimap.py → alfie/trimap.py b/sam_aclip_pixart_sigma/trimap.py → alfie/trimap.py
diff --git a/sam_aclip_pixart_sigma/utils.py → alfie/utils.py b/sam_aclip_pixart_sigma/utils.py → alfie/utils.py
diff --git 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