Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -318,7 +318,6 @@ def worker(input_image, image_position, prompts, n_prompt, seed, resolution, tot
|
|
| 318 |
|
| 319 |
llama_vec, llama_attention_mask = crop_or_pad_yield_mask(llama_vec, length=512)
|
| 320 |
llama_vec_n, llama_attention_mask_n = crop_or_pad_yield_mask(llama_vec_n, length=512)
|
| 321 |
-
print("llama_attention_mask_n shape: " + str(llama_attention_mask_n.shape))
|
| 322 |
|
| 323 |
llama_vec = llama_vec.to(transformer.dtype)
|
| 324 |
llama_vec_n = llama_vec_n.to(transformer.dtype)
|
|
|
|
| 318 |
|
| 319 |
llama_vec, llama_attention_mask = crop_or_pad_yield_mask(llama_vec, length=512)
|
| 320 |
llama_vec_n, llama_attention_mask_n = crop_or_pad_yield_mask(llama_vec_n, length=512)
|
|
|
|
| 321 |
|
| 322 |
llama_vec = llama_vec.to(transformer.dtype)
|
| 323 |
llama_vec_n = llama_vec_n.to(transformer.dtype)
|