willsh1997 commited on
Commit
8b827ee
·
verified ·
1 Parent(s): 538d863
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -419,9 +419,8 @@ will_cand_tensors = torch.cat([chaosclicker_willtensor,
419
  @spaces.GPU
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
 
427
  ### GRADIO BACKEND
 
419
  @spaces.GPU
420
  def generate_freak():
421
  will_randomised_input = random_candtensor(will_cand_tensors).unsqueeze(0)
422
+ #will_randomised_input
423
+ output = pipe(image_embeddings=will_randomised_input.to("cuda"), num_images_per_prompt=1, decoder_num_inference_steps = 15, super_res_num_inference_steps = 4)
 
424
  return output.images[0]
425
 
426
  ### GRADIO BACKEND