Update app.py

app.py

@@ -44,13 +44,21 @@ tokenizer = CLIPTokenizer.from_pretrained(
 tokenizer_2 = CLIPTokenizer.from_pretrained(
     sdxl_name, subfolder="tokenizer_2")
 text_encoder = CLIPTextModel.from_pretrained(
-    sdxl_name, subfolder="text_encoder", torch_dtype=torch.float16, variant="fp16", device_map="auto")
+    sdxl_name, subfolder="text_encoder", torch_dtype=torch.float16, 
+    #variant="fp16",
+    device_map="auto")
 text_encoder_2 = CLIPTextModel.from_pretrained(
-    sdxl_name, subfolder="text_encoder_2", torch_dtype=torch.float16, variant="fp16", device_map="auto")
+    sdxl_name, subfolder="text_encoder_2", torch_dtype=torch.float16, 
+    #variant="fp16",
+    device_map="auto")
 vae = AutoencoderKL.from_pretrained(
-    sdxl_name, subfolder="vae", torch_dtype=torch.bfloat16, variant="fp16", device_map="auto")  # bfloat16 vae
+    sdxl_name, subfolder="vae", torch_dtype=torch.bfloat16, 
+    #variant="fp16",
+    device_map="auto")  # bfloat16 vae
 unet = UNet2DConditionModel.from_pretrained(
-    sdxl_name, subfolder="unet", torch_dtype=torch.float16, variant="fp16", device_map="auto")
+    sdxl_name, subfolder="unet", torch_dtype=torch.float16, 
+    #variant="fp16",
+    device_map="auto")
 
 unet.set_attn_processor(AttnProcessor2_0())
 vae.set_attn_processor(AttnProcessor2_0())