StonyBrook-CVLab
/

PixCell-256

PixCellPipeline

Model card Files Files and versions

AlexGraikos commited on Apr 9, 2025

Commit

462ac0e

·

verified ·

1 Parent(s): 8fb01de

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -16,9 +16,9 @@ device = torch.device('cuda')
 sd3_vae = AutoencoderKL.from_pretrained("stabilityai/stable-diffusion-3.5-large", subfolder="vae")
 pipeline = DiffusionPipeline.from_pretrained(
-    "StonyBrook-CVLab/pixcell-256-diffusers",
     vae=sd3_vae,
-    custom_pipeline="StonyBrook-CVLab/pixcell-pipeline",
     trust_remote_code=True,
 )
@@ -66,7 +66,7 @@ from PIL import Image
 from huggingface_hub import hf_hub_download
 # This is an example image we provide
-path = hf_hub_download(repo_id="StonyBrook-CVLab/pixcell-256-diffusers", filename="test_image.jpg")
 image = Image.open(path)
 # Extract UNI from random patches in the image

 sd3_vae = AutoencoderKL.from_pretrained("stabilityai/stable-diffusion-3.5-large", subfolder="vae")
 pipeline = DiffusionPipeline.from_pretrained(
+    "StonyBrook-CVLab/PixCell-256",
     vae=sd3_vae,
+    custom_pipeline="StonyBrook-CVLab/PixCell-pipeline",
     trust_remote_code=True,
 )
 from huggingface_hub import hf_hub_download
 # This is an example image we provide
+path = hf_hub_download(repo_id="StonyBrook-CVLab/PixCell-256", filename="test_image.jpg")
 image = Image.open(path)
 # Extract UNI from random patches in the image