Spaces:
Paused
Paused
Upload folder using huggingface_hub
Browse files
backend/lcm_text_to_image.py
CHANGED
|
@@ -80,6 +80,13 @@ class LCMTextToImage:
|
|
| 80 |
beta_end=0.01,
|
| 81 |
)
|
| 82 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 83 |
def init(
|
| 84 |
self,
|
| 85 |
device: str = "cpu",
|
|
@@ -97,7 +104,7 @@ class LCMTextToImage:
|
|
| 97 |
if lcm_diffusion_setting.diffusion_task == DiffusionTask.image_to_image.value:
|
| 98 |
w, h = lcm_diffusion_setting.init_image.size
|
| 99 |
newW = lcm_diffusion_setting.image_width
|
| 100 |
-
newH = int(h * newW / w)
|
| 101 |
lcm_diffusion_setting.image_height=newH
|
| 102 |
img = lcm_diffusion_setting.init_image.resize((newW, newH))
|
| 103 |
print("新图",newH,newW, lcm_diffusion_setting.image_height)
|
|
|
|
| 80 |
beta_end=0.01,
|
| 81 |
)
|
| 82 |
|
| 83 |
+
def make_even(self,num):
|
| 84 |
+
if num % 2 == 0:
|
| 85 |
+
return num
|
| 86 |
+
else:
|
| 87 |
+
print("取整了")
|
| 88 |
+
return math.ceil(num)
|
| 89 |
+
|
| 90 |
def init(
|
| 91 |
self,
|
| 92 |
device: str = "cpu",
|
|
|
|
| 104 |
if lcm_diffusion_setting.diffusion_task == DiffusionTask.image_to_image.value:
|
| 105 |
w, h = lcm_diffusion_setting.init_image.size
|
| 106 |
newW = lcm_diffusion_setting.image_width
|
| 107 |
+
newH = self.make_even(int(h * newW / w))
|
| 108 |
lcm_diffusion_setting.image_height=newH
|
| 109 |
img = lcm_diffusion_setting.init_image.resize((newW, newH))
|
| 110 |
print("新图",newH,newW, lcm_diffusion_setting.image_height)
|