Spaces:
Runtime error
Runtime error
version 2
Browse files
app.py
CHANGED
|
@@ -33,7 +33,6 @@ def predict(img, strength, seed, prompt):
|
|
| 33 |
|
| 34 |
init_image = img2.resize((768, 512))
|
| 35 |
|
| 36 |
-
|
| 37 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 38 |
with autocast("cuda"):
|
| 39 |
image = pipe(prompt=prompt, init_image=init_image, strength=strength, guidance_scale=5, generator=generator).images[0]
|
|
|
|
| 33 |
|
| 34 |
init_image = img2.resize((768, 512))
|
| 35 |
|
|
|
|
| 36 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 37 |
with autocast("cuda"):
|
| 38 |
image = pipe(prompt=prompt, init_image=init_image, strength=strength, guidance_scale=5, generator=generator).images[0]
|