Upload folder using huggingface_hub

DiT-XL-2-256/README.md

@@ -27,6 +27,7 @@ pipe = DiffusionPipeline.from_pretrained(
     str(model_dir),
     local_files_only=True,
     custom_pipeline=str(model_dir / "pipeline.py"),
+    trust_remote_code=True,
     torch_dtype=torch.float32,
 ).to("cuda" if torch.cuda.is_available() else "cpu")
 

DiT-XL-2-256/pipeline.py

@@ -35,6 +35,7 @@ EXAMPLE_DOC_STRING = """
         ...     str(model_dir),
         ...     local_files_only=True,
         ...     custom_pipeline=str(model_dir / "pipeline.py"),
+        ...     trust_remote_code=True,
         ...     torch_dtype=torch.float16,
         ... )
         >>> pipe = pipe.to("cuda")

DiT-XL-2-512/README.md

@@ -58,6 +58,7 @@ pipe = DiffusionPipeline.from_pretrained(
     str(model_dir),
     local_files_only=True,
     custom_pipeline=str(model_dir / "pipeline.py"),
+    trust_remote_code=True,
     torch_dtype=torch.float32,
 ).to("cuda")
 

DiT-XL-2-512/pipeline.py

@@ -35,6 +35,7 @@ EXAMPLE_DOC_STRING = """
         ...     str(model_dir),
         ...     local_files_only=True,
         ...     custom_pipeline=str(model_dir / "pipeline.py"),
+        ...     trust_remote_code=True,
         ...     torch_dtype=torch.float16,
         ... )
         >>> pipe = pipe.to("cuda")

README.md

@@ -60,6 +60,7 @@ pipe = DiffusionPipeline.from_pretrained(
     str(model_dir),
     local_files_only=True,
     custom_pipeline=str(model_dir / "pipeline.py"),
+    trust_remote_code=True,
     torch_dtype=torch.bfloat16,
 ).to("cuda")
 generator = torch.Generator(device="cuda").manual_seed(0)