.

scripts/anime.py

@@ -25,7 +25,6 @@ def init_model(use_local=False):
     model.eval()
 
 # numpy配列の画像を受け取り、線画を生成してnumpy配列で返す
-# @spaces.GPU
 def generate_sketch(image, clahe_clip=-1, load_size=512):
     """
     Generate sketch image from input image
@@ -67,7 +66,6 @@ def generate_sketch(image, clahe_clip=-1, load_size=512):
     aus_img = tensor_to_img(aus_tensor)
     return aus_img
 
-
 if __name__ == '__main__':
     os.chdir(os.path.dirname("Anime2Sketch/"))
     parser = argparse.ArgumentParser(description='Anime-to-sketch test options.')

scripts/process_utils.py

@@ -17,7 +17,7 @@ from scripts.hf_utils import download_file
 use_local = False
 model = None
 device = None
-torch_dtype = None # torch.float16 if device == "cuda" else torch.float32
+torch_dtype = None
 sotai_gen_pipe = None
 refine_gen_pipe = None
 
@@ -43,7 +43,6 @@ def initialize(_use_local=False, use_gpu=False, use_dotenv=False):
     print(f"\nDevice: {device}, Local model: {_use_local}\n")
 
     init_model(use_local)
-    # model = load_wd14_tagger_model()
     sotai_gen_pipe = initialize_sotai_model()
     refine_gen_pipe = initialize_refine_model()
 
@@ -201,7 +200,6 @@ def create_rgba_image(binary_image: np.ndarray, color: list) -> Image.Image:
     rgba_image[:, :, 3] = binary_image
     return Image.fromarray(rgba_image, 'RGBA')
 
-# @spaces.GPU
 def generate_sotai_image(input_image: Image.Image, output_width: int, output_height: int) -> Image.Image:
     input_image = ensure_rgb(input_image)
     global sotai_gen_pipe
@@ -246,7 +244,6 @@ def generate_sotai_image(input_image: Image.Image, output_width: int, output_hei
             torch.cuda.empty_cache()
         gc.collect()
 
-# @spaces.GPU
 def generate_refined_image(prompt: str, original_image: Image.Image, output_width: int, output_height: int, weight1: float, weight2: float) -> Image.Image:
     original_image = ensure_rgb(original_image)
     global refine_gen_pipe