MaxMilan1
commited on
Commit
·
ee45526
1
Parent(s):
6309bfe
changees
Browse files- util/text_img.py +1 -1
util/text_img.py
CHANGED
|
@@ -58,7 +58,7 @@ def generate_imgtoimg(prompt, init_image, strength=0.5):
|
|
| 58 |
prompt += ", no background, side view, minimalist shot, single shoe, no legs, product photo"
|
| 59 |
|
| 60 |
imagepipe = AutoPipelineForImage2Image.from_pretrained(
|
| 61 |
-
"stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.
|
| 62 |
)
|
| 63 |
|
| 64 |
image = imagepipe(prompt, image=init_image, strength=strength).images[0]
|
|
|
|
| 58 |
prompt += ", no background, side view, minimalist shot, single shoe, no legs, product photo"
|
| 59 |
|
| 60 |
imagepipe = AutoPipelineForImage2Image.from_pretrained(
|
| 61 |
+
"stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float32, use_safetensors=True
|
| 62 |
)
|
| 63 |
|
| 64 |
image = imagepipe(prompt, image=init_image, strength=strength).images[0]
|