mlpc-lab
/

TokenCompose_SD21_B

StableDiffusionPipeline

stable-diffusion

image-generation

compositionality

Model card Files Files and versions

JamesSand commited on Dec 4, 2023

Commit

1627bf0

·

1 Parent(s): 746ec97

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_B"
 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_B"
 device = "cuda"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
 pipe = pipe.to(device)
 prompt = "A cat and a wine glass"