Hugging Face's logo

Spaces:
gaur3009
/
Text2img
Running

App Files Community
gaur3009 commited on
Commit
a9465a8
·
verified ·
1 Parent(s): 2952a26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from PIL import Image
5
 
6
  # Load Stable Diffusion XL base model with LoRA weights (requires PEFT backend)
7
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
8
- pipe.to("cuda")
9
 
10
  # Ensure PEFT is installed and compatible
11
  try:
 
5
 
6
  # Load Stable Diffusion XL base model with LoRA weights (requires PEFT backend)
7
  pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
8
+ pipe.to("cpu")
9
 
10
  # Ensure PEFT is installed and compatible
11
  try: