retwpay commited on
Commit
4f5fe11
·
verified ·
1 Parent(s): 480ec3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
12
 
13
  # Make sure to use torch.float16 consistently throughout the pipeline
14
  pipe = StableDiffusionXLPipeline.from_pretrained(
15
- "votepurchase/waiNSFWIllustrious_v110",
16
  torch_dtype=torch.float16,
17
  variant="fp16", # Explicitly use fp16 variant
18
  use_safetensors=True # Use safetensors if available
 
12
 
13
  # Make sure to use torch.float16 consistently throughout the pipeline
14
  pipe = StableDiffusionXLPipeline.from_pretrained(
15
+ "votepurchase/waiNSFWIllustrious_v120",
16
  torch_dtype=torch.float16,
17
  variant="fp16", # Explicitly use fp16 variant
18
  use_safetensors=True # Use safetensors if available