nicolas-dufour
/

miro-ablations

reward-conditioning

Model card Files Files and versions

nicolas-dufour commited on 10 days ago

Commit

4862325

·

verified ·

1 Parent(s): f2e0f43

Upload miro-only-aesthetic weights

Files changed (1) hide show

miro-only-aesthetic/README.md +1 -1

miro-only-aesthetic/README.md CHANGED Viewed

@@ -57,7 +57,7 @@ from miro import MiroPipeline
 pipe = MiroPipeline.from_pretrained(
     "nicolas-dufour/miro-ablations", variant="miro-only-aesthetic",
 )
-pipe = pipe.to("cuda", torch.float16)
 prompt = (
     "Photography closeup portrait of an adorable rusty brokendown steampunk "

 pipe = MiroPipeline.from_pretrained(
     "nicolas-dufour/miro-ablations", variant="miro-only-aesthetic",
 )
+pipe = pipe.to("cuda", torch.bfloat16)
 prompt = (
     "Photography closeup portrait of an adorable rusty brokendown steampunk "