Update app.py
Browse files
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 |
-
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
|
| 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')
|