Hugging Face's logo

Spaces:
AP123
/
CerealBoxMaker
Runtime error

App Files Community
1
AP123 commited on
Commit
1e875da
·
1 Parent(s): 3f6c8b1

fp 16 to allow for T4 inference

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from PIL import Image
5
  import random
6
  from diffusers import DiffusionPipeline
7
 
8
- pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
9
  pipeline.load_lora_weights("ostris/super-cereal-sdxl-lora")
10
  pipeline.to("cuda:0")
11
 
 
5
  import random
6
  from diffusers import DiffusionPipeline
7
 
8
+ pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16")
9
  pipeline.load_lora_weights("ostris/super-cereal-sdxl-lora")
10
  pipeline.to("cuda:0")
11