Spaces:
Running
Running
moving pipe to cuda in desparate position lol
Browse files
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 |
|