app.py
CHANGED
|
@@ -32,7 +32,7 @@ class Pipe:
|
|
| 32 |
h, w = image_tensor.shape[-2], image_tensor.shape[-1]
|
| 33 |
h, w = h // 32 * 32, w // 32 * 32
|
| 34 |
image_tensor = torch.nn.functional.interpolate(image_tensor, size=(h, w), mode='bilinear')
|
| 35 |
-
ipt=self.net_c(image_tensor)
|
| 36 |
image_tensor = image_tensor.half()
|
| 37 |
ipt = ipt.half()
|
| 38 |
output_i, output_j=self.net_i(image_tensor,ipt,prompt=True)
|
|
|
|
| 32 |
h, w = image_tensor.shape[-2], image_tensor.shape[-1]
|
| 33 |
h, w = h // 32 * 32, w // 32 * 32
|
| 34 |
image_tensor = torch.nn.functional.interpolate(image_tensor, size=(h, w), mode='bilinear')
|
| 35 |
+
ipt=self.net_c(image_tensor.cuda())
|
| 36 |
image_tensor = image_tensor.half()
|
| 37 |
ipt = ipt.half()
|
| 38 |
output_i, output_j=self.net_i(image_tensor,ipt,prompt=True)
|