afeng commited on
Commit
56db6e4
1 Parent(s): 8963af6
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  import numpy as np
4
  import argparse
5
  from peft import LoraConfig
6
- from old.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
 
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