Spaces:
Running
on
A10G
Running
on
A10G
Linoy Tsaban
commited on
Commit
·
86ca2b5
1
Parent(s):
02f2630
Update app.py
Browse files
app.py
CHANGED
|
@@ -105,8 +105,8 @@ def edit(input_image,
|
|
| 105 |
|
| 106 |
cfg_scales_tensor = torch.Tensor(cfg_scales).view(-1,1,1,1).to(sd_pipe.device)
|
| 107 |
|
| 108 |
-
text_embeddings = encode_text(
|
| 109 |
-
uncond_embedding = encode_text(
|
| 110 |
|
| 111 |
if etas is None: etas = 0
|
| 112 |
if type(etas) in [int, float]: etas = [etas]*sd_pipe.scheduler.num_inference_steps
|
|
|
|
| 105 |
|
| 106 |
cfg_scales_tensor = torch.Tensor(cfg_scales).view(-1,1,1,1).to(sd_pipe.device)
|
| 107 |
|
| 108 |
+
text_embeddings = encode_text(sd_pipe, prompts)
|
| 109 |
+
uncond_embedding = encode_text(sd_pipe, [""] * batch_size)
|
| 110 |
|
| 111 |
if etas is None: etas = 0
|
| 112 |
if type(etas) in [int, float]: etas = [etas]*sd_pipe.scheduler.num_inference_steps
|