Spaces:

Abiru
/

UnfilteredAI-NSFW-gen-v2

Running

App Files Files Community

Abiru commited on Jul 3

Commit

af69fa5

•

1 Parent(s): 7d62036

Update app.py

Files changed (1) hide show

app.py +1 -0

app.py CHANGED Viewed

@@ -4,6 +4,7 @@ import random
 import spaces
 pipe = DiffusionPipeline.from_pretrained("UnfilteredAI/NSFW-gen-v2")
 @spaces.GPU
 def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width, height, num_samples):

 import spaces
 pipe = DiffusionPipeline.from_pretrained("UnfilteredAI/NSFW-gen-v2")
+pipe.to('cuda')
 @spaces.GPU
 def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width, height, num_samples):