mlpc-lab
/

TokenCompose_SD21_A

StableDiffusionPipeline

stable-diffusion

image-generation

compositionality

Inference Endpoints

Model card Files Files and versions Community

JamesSand commited on Dec 4, 2023

Commit

80a154c

•

1 Parent(s): 6f0b189

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -22,7 +22,7 @@ from diffusers import StableDiffusionPipeline
 model_id = "mlpc-lab/TokenCompose_SD21_A"
 device = "cuda"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe = pipe.to(device)
 prompt = "A cat and a wine glass"

 model_id = "mlpc-lab/TokenCompose_SD21_A"
 device = "cuda"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
 pipe = pipe.to(device)
 prompt = "A cat and a wine glass"