prithivMLmods commited on
Commit
2d98ba3
1 Parent(s): 44ccefa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -41,16 +41,15 @@ NUM_IMAGES_PER_PROMPT = 1
41
 
42
  if torch.cuda.is_available():
43
  pipe = DiffusionPipeline.from_pretrained(
44
- "SG161222/RealVisXL_V4.0_Lightning",
45
  ##"SG161222/RealVisXL_V1.02_Turbo",
46
- ##"SG161222/RealVisXL_V3.0",
47
  torch_dtype=torch.float16,
48
  use_safetensors=True,
49
  add_watermarker=False,
50
  variant="fp16"
51
  )
52
  pipe2 = DiffusionPipeline.from_pretrained(
53
- "SG161222/RealVisXL_V4.0",
54
  torch_dtype=torch.float16,
55
  use_safetensors=True,
56
  add_watermarker=False,
 
41
 
42
  if torch.cuda.is_available():
43
  pipe = DiffusionPipeline.from_pretrained(
44
+ "SG161222/RealVisXL_V4.0",
45
  ##"SG161222/RealVisXL_V1.02_Turbo",
 
46
  torch_dtype=torch.float16,
47
  use_safetensors=True,
48
  add_watermarker=False,
49
  variant="fp16"
50
  )
51
  pipe2 = DiffusionPipeline.from_pretrained(
52
+ "SG161222/RealVisXL_V3.0",
53
  torch_dtype=torch.float16,
54
  use_safetensors=True,
55
  add_watermarker=False,