Fabrice-TIERCELIN commited on
Commit
80ffd55
·
verified ·
1 Parent(s): 4d748d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
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)