second
Browse files
main.py
CHANGED
@@ -3,7 +3,7 @@ import torch
|
|
3 |
import numpy as np
|
4 |
import argparse
|
5 |
from peft import LoraConfig
|
6 |
-
from
|
7 |
from pipeline_dedit_sd import DEditSDPipeline
|
8 |
from utils import load_image, load_mask, load_mask_edit
|
9 |
from utils_mask import process_mask_move_torch, process_mask_remove_torch, mask_union_torch, mask_substract_torch, create_outer_edge_mask_torch
|
|
|
3 |
import numpy as np
|
4 |
import argparse
|
5 |
from peft import LoraConfig
|
6 |
+
from pipeline_dedit_sdxl import DEditSDXLPipeline
|
7 |
from pipeline_dedit_sd import DEditSDPipeline
|
8 |
from utils import load_image, load_mask, load_mask_edit
|
9 |
from utils_mask import process_mask_move_torch, process_mask_remove_torch, mask_union_torch, mask_substract_torch, create_outer_edge_mask_torch
|