David Burnett
commited on
Commit
•
540d89c
1
Parent(s):
d69a081
update readme again
Browse files
README.md
CHANGED
|
@@ -28,7 +28,7 @@ device = 'mps'
|
|
| 28 |
weight_dtype = torch.bfloat16
|
| 29 |
|
| 30 |
pipe = PixArtSigmaPipeline.from_pretrained(
|
| 31 |
-
"/
|
| 32 |
torch_dtype=weight_dtype,
|
| 33 |
variant="fp16",
|
| 34 |
use_safetensors=True,
|
|
@@ -45,8 +45,8 @@ for i in range(4):
|
|
| 45 |
seed = random.randint(0, sys.maxsize)
|
| 46 |
generator = torch.Generator("mps").manual_seed(seed);
|
| 47 |
|
| 48 |
-
|
| 49 |
image = pipe(prompt, generator=generator, num_iferencenum_inference_steps=40).images[0]
|
| 50 |
image.save(f"pas_{seed}.png")a
|
|
|
|
| 51 |
```
|
| 52 |
|
|
|
|
| 28 |
weight_dtype = torch.bfloat16
|
| 29 |
|
| 30 |
pipe = PixArtSigmaPipeline.from_pretrained(
|
| 31 |
+
"/Vargol/PixArt-Sigma_16bit",
|
| 32 |
torch_dtype=weight_dtype,
|
| 33 |
variant="fp16",
|
| 34 |
use_safetensors=True,
|
|
|
|
| 45 |
seed = random.randint(0, sys.maxsize)
|
| 46 |
generator = torch.Generator("mps").manual_seed(seed);
|
| 47 |
|
|
|
|
| 48 |
image = pipe(prompt, generator=generator, num_iferencenum_inference_steps=40).images[0]
|
| 49 |
image.save(f"pas_{seed}.png")a
|
| 50 |
+
|
| 51 |
```
|
| 52 |
|