rahul7star commited on
Commit
c5f7d50
·
verified ·
1 Parent(s): 9418431

Update app_t2v.py

Browse files
Files changed (1) hide show
  1. app_t2v.py +10 -10
app_t2v.py CHANGED
@@ -29,17 +29,8 @@ print("Loading Stable Diffusion XL model...")
29
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
30
 
31
  # Initialize SDXL pipeline
32
- # sdxl_pipe = StableDiffusionXLPipeline.from_pretrained(
33
- # "votepurchase/pornmasterPro_noobV3VAE",
34
- # torch_dtype=torch.float16,
35
- # variant="fp16",
36
- # use_safetensors=True
37
- # )
38
-
39
-
40
-
41
  sdxl_pipe = StableDiffusionXLPipeline.from_pretrained(
42
- "stablediffusionapi/omnigenxl-nsfw-sfw",
43
  torch_dtype=torch.float16,
44
  variant="fp16",
45
  use_safetensors=True
@@ -47,6 +38,15 @@ sdxl_pipe = StableDiffusionXLPipeline.from_pretrained(
47
 
48
 
49
 
 
 
 
 
 
 
 
 
 
50
 
51
 
52
  sdxl_pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(sdxl_pipe.scheduler.config)
 
29
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
30
 
31
  # Initialize SDXL pipeline
 
 
 
 
 
 
 
 
 
32
  sdxl_pipe = StableDiffusionXLPipeline.from_pretrained(
33
+ "votepurchase/pornmasterPro_noobV3VAE",
34
  torch_dtype=torch.float16,
35
  variant="fp16",
36
  use_safetensors=True
 
38
 
39
 
40
 
41
+ # sdxl_pipe = StableDiffusionXLPipeline.from_pretrained(
42
+ # "stablediffusionapi/omnigenxl-nsfw-sfw",
43
+ # torch_dtype=torch.float16,
44
+ # variant="fp16",
45
+ # use_safetensors=True
46
+ # )
47
+
48
+
49
+
50
 
51
 
52
  sdxl_pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(sdxl_pipe.scheduler.config)