From 1ab3eb0517c8600c219f0994ace251cf8788d2bf Mon Sep 17 00:00:00 2001 From: quattrinifabio Date: Thu, 8 Aug 2024 14:51:56 +0200 Subject: [PATCH] refactoring --- {sam_aclip_pixart_sigma => alfie}/__init__.py | 0 {sam_aclip_pixart_sigma => alfie}/attn_processor.py | 0 {sam_aclip_pixart_sigma => alfie}/cross_heatmap.py | 0 {sam_aclip_pixart_sigma => alfie}/generate.py | 0 {sam_aclip_pixart_sigma => alfie}/grabcut.py | 0 .../pipeline_pixart_sigma.py | 0 {sam_aclip_pixart_sigma => alfie}/self_heatmap.py | 0 {sam_aclip_pixart_sigma => alfie}/transformer_2d.py | 0 {sam_aclip_pixart_sigma => alfie}/trimap.py | 0 {sam_aclip_pixart_sigma => alfie}/utils.py | 0 generate_prompt.py | 8 ++++---- 11 files changed, 4 insertions(+), 4 deletions(-) rename {sam_aclip_pixart_sigma => alfie}/__init__.py (100%) rename {sam_aclip_pixart_sigma => alfie}/attn_processor.py (100%) rename {sam_aclip_pixart_sigma => alfie}/cross_heatmap.py (100%) rename {sam_aclip_pixart_sigma => alfie}/generate.py (100%) rename {sam_aclip_pixart_sigma => alfie}/grabcut.py (100%) rename {sam_aclip_pixart_sigma => alfie}/pipeline_pixart_sigma.py (100%) rename {sam_aclip_pixart_sigma => alfie}/self_heatmap.py (100%) rename {sam_aclip_pixart_sigma => alfie}/transformer_2d.py (100%) rename {sam_aclip_pixart_sigma => alfie}/trimap.py (100%) rename {sam_aclip_pixart_sigma => alfie}/utils.py (100%) 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