Spaces:

Collov-Labs
/

d-edit

Running on Zero

afeng commited on Sep 4

Commit

56db6e4

•

1 Parent(s): 8963af6

second

Files changed (1) hide show

main.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 import numpy as np
 import argparse
 from peft import LoraConfig
-from old.pipeline_dedit_sdxl import DEditSDXLPipeline
 from pipeline_dedit_sd import DEditSDPipeline
 from utils import load_image, load_mask, load_mask_edit
 from utils_mask import process_mask_move_torch, process_mask_remove_torch, mask_union_torch, mask_substract_torch, create_outer_edge_mask_torch

 import numpy as np
 import argparse
 from peft import LoraConfig
+from pipeline_dedit_sdxl import DEditSDXLPipeline
 from pipeline_dedit_sd import DEditSDPipeline
 from utils import load_image, load_mask, load_mask_edit
 from utils_mask import process_mask_move_torch, process_mask_remove_torch, mask_union_torch, mask_substract_torch, create_outer_edge_mask_torch