diff --git a/gradio_generate.py b/gradio_generate.py index 9d18d94..a9b75ab 100644 --- a/gradio_generate.py +++ b/gradio_generate.py @@ -6,7 +6,7 @@ from garment_adapter.garment_diffusion import ClothAdapter from pipelines.OmsDiffusionPipeline import OmsDiffusionPipeline -from pipelines.OmsIndependentDiffusionPipeline import OmsIndependentDiffusionPipeline +#from pipelines.OmsIndependentDiffusionPipeline import OmsIndependentDiffusionPipeline parser = argparse.ArgumentParser(description='oms diffusion') parser.add_argument('--model_path', type=str, required=True)