1inkusFace commited on
Commit
a223986
·
verified ·
1 Parent(s): 859687d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -118,10 +118,10 @@ model_repo='John6666/uber-realistic-porn-merge-xl-urpmxl-v6final-sdxl'
118
 
119
  rv='ford442/RealVisXL_V5.0_BF16'
120
 
121
- #text_encoder = CLIPTextModel.from_pretrained(rv, subfolder='text_encoder')#.to(device=device, dtype=torch.bfloat16)
122
- #text_encoder_2 = CLIPTextModelWithProjection.from_pretrained(model_repo, subfolder='text_encoder_2')#.to(device=device, dtype=torch.bfloat16)
123
- #tokenizer_1 = CLIPTokenizer.from_pretrained(rv, subfolder='tokenizer')
124
- #tokenizer_2 = CLIPTokenizer.from_pretrained(model_repo, subfolder='tokenizer_2')
125
  #scheduler = EulerAncestralDiscreteScheduler.from_pretrained(rv, subfolder='scheduler')
126
  vaeXL = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", low_cpu_mem_usage=False, safety_checker=None, use_safetensors=False, torch_dtype=torch.float32) #.to(device).to(torch.bfloat16) #.to(device=device, dtype=torch.bfloat16)
127
  #unet = UNet2DConditionModel.from_pretrained(model_repo, low_cpu_mem_usage=False, subfolder='unet', upcast_attention=True, attention_type='gated-text-image')
 
118
 
119
  rv='ford442/RealVisXL_V5.0_BF16'
120
 
121
+ text_encoder = CLIPTextModel.from_pretrained(rv, subfolder='text_encoder')#.to(device=device, dtype=torch.bfloat16)
122
+ text_encoder_2 = CLIPTextModelWithProjection.from_pretrained(model_repo, subfolder='text_encoder_2')#.to(device=device, dtype=torch.bfloat16)
123
+ tokenizer_1 = CLIPTokenizer.from_pretrained(rv, subfolder='tokenizer')
124
+ tokenizer_2 = CLIPTokenizer.from_pretrained(model_repo, subfolder='tokenizer_2')
125
  #scheduler = EulerAncestralDiscreteScheduler.from_pretrained(rv, subfolder='scheduler')
126
  vaeXL = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", low_cpu_mem_usage=False, safety_checker=None, use_safetensors=False, torch_dtype=torch.float32) #.to(device).to(torch.bfloat16) #.to(device=device, dtype=torch.bfloat16)
127
  #unet = UNet2DConditionModel.from_pretrained(model_repo, low_cpu_mem_usage=False, subfolder='unet', upcast_attention=True, attention_type='gated-text-image')