dhead commited on
Commit
a9c8bb8
·
verified ·
1 Parent(s): 7902330

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
  pipe = StableDiffusionXLPipeline.from_pretrained(
18
  "dhead/ntr-mix-illustrious-xl-noob-xl-xiii-sdxl",
19
  torch_dtype=torch.float16,
20
- variant="fp16", # Explicitly use fp16 variant
21
  use_safetensors=True # Use safetensors if available
22
  )
23
 
 
17
  pipe = StableDiffusionXLPipeline.from_pretrained(
18
  "dhead/ntr-mix-illustrious-xl-noob-xl-xiii-sdxl",
19
  torch_dtype=torch.float16,
20
+ # variant="fp16", # Explicitly use fp16 variant
21
  use_safetensors=True # Use safetensors if available
22
  )
23