Upload folder using huggingface_hub

README.md

@@ -67,6 +67,7 @@ model_dir = Path("./JiT-H-32")
 pipe = DiffusionPipeline.from_pretrained(
     str(model_dir),
     custom_pipeline=str(model_dir / "pipeline.py"),
+    trust_remote_code=True,
 )
 pipe.scheduler = FlowMatchHeunDiscreteScheduler.from_config(pipe.scheduler.config, shift=4.0)
 pipe.to("cuda")

demo_inference.py

@@ -15,10 +15,12 @@ def main() -> None:
     pipe = DiffusionPipeline.from_pretrained(
         str(MODEL_DIR),
         custom_pipeline=str(MODEL_DIR / "pipeline.py"),
+        trust_remote_code=True,
         torch_dtype=torch.bfloat16,
     )
     pipe.scheduler = FlowMatchHeunDiscreteScheduler.from_config(pipe.scheduler.config, shift=4.0)
     pipe.to("cuda")
+    pipe.set_progress_bar_config(disable=False)
 
     print(pipe.id2label[207])
     print(pipe.get_label_ids("golden retriever"))