willsh1997 commited on
Commit
538d863
·
verified ·
1 Parent(s): d45dbc3

moving pipe to cuda in desparate position lol

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -420,6 +420,7 @@ will_cand_tensors = torch.cat([chaosclicker_willtensor,
420
  def generate_freak():
421
  will_randomised_input = random_candtensor(will_cand_tensors).unsqueeze(0)
422
  will_randomised_input.to("cuda")
 
423
  output = pipe(image_embeddings=will_randomised_input, num_images_per_prompt=1, decoder_num_inference_steps = 15, super_res_num_inference_steps = 4)
424
  return output.images[0]
425
 
 
420
  def generate_freak():
421
  will_randomised_input = random_candtensor(will_cand_tensors).unsqueeze(0)
422
  will_randomised_input.to("cuda")
423
+ pipe.to("cuda") #a real hail mary here
424
  output = pipe(image_embeddings=will_randomised_input, num_images_per_prompt=1, decoder_num_inference_steps = 15, super_res_num_inference_steps = 4)
425
  return output.images[0]
426