Spaces:
Running
on
T4
Running
on
T4
wider width image test
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float
|
|
| 12 |
pipe = pipe.to("cuda")
|
| 13 |
|
| 14 |
def predict(prompt):
|
| 15 |
-
spec = pipe(prompt).images[0]
|
| 16 |
print(spec)
|
| 17 |
wav = wav_bytes_from_spectrogram_image(spec)
|
| 18 |
with open("output.wav", "wb") as f:
|
|
|
|
| 12 |
pipe = pipe.to("cuda")
|
| 13 |
|
| 14 |
def predict(prompt):
|
| 15 |
+
spec = pipe(prompt, width=768).images[0]
|
| 16 |
print(spec)
|
| 17 |
wav = wav_bytes_from_spectrogram_image(spec)
|
| 18 |
with open("output.wav", "wb") as f:
|