Spaces:

Ghamryy
/

Jamal

Paused

Ghamryy commited on Jun 30, 2024

Commit

ef75de5

•

1 Parent(s): ecf15a7

cuda enabled

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ base = DiffusionPipeline.from_pretrained(
     torch_dtype=torch.float16,
     use_safetensors=True
 )
-# base.to("cuda")
 def text2Image(prompt, steps=50, scale=7, Width=1024, Height=1024):
     image = base(prompt, num_inference_steps=steps, guidance_scale=scale,

     torch_dtype=torch.float16,
     use_safetensors=True
 )
+base.to("cuda")
 def text2Image(prompt, steps=50, scale=7, Width=1024, Height=1024):
     image = base(prompt, num_inference_steps=steps, guidance_scale=scale,