Update app.py

app.py

@@ -182,8 +182,8 @@ def generate_30(
 ):
     seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device='cuda').manual_seed(seed)
-    pipe.text_encoder=CLIPTextModel.from_config(pipe.config)
-    pipe.text_encoder_2=CLIPTextModelWithProjection.from_config(pipe.config)
+    pipe.text_encoder=CLIPTextModel.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='text_encoder',token=True).to(device=device, dtype=torch.bfloat16)
+    pipe.text_encoder_2=CLIPTextModelWithProjection.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='text_encoder_2',token=True).to(device=device, dtype=torch.bfloat16)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],
@@ -224,8 +224,8 @@ def generate_60(
 ):
     seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device='cuda').manual_seed(seed)
-    pipe.text_encoder=CLIPTextModel.from_config(pipe.config)
-    pipe.text_encoder_2=CLIPTextModelWithProjection.from_config(pipe.config)
+    pipe.text_encoder=CLIPTextModel.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='text_encoder',token=True).to(device=device, dtype=torch.bfloat16)
+    pipe.text_encoder_2=CLIPTextModelWithProjection.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='text_encoder_2',token=True).to(device=device, dtype=torch.bfloat16)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],
@@ -266,8 +266,8 @@ def generate_90(
 ):
     seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device='cuda').manual_seed(seed)
-    pipe.text_encoder=CLIPTextModel.from_config(pipe.config)
-    pipe.text_encoder_2=CLIPTextModelWithProjection.from_config(pipe.config)
+    pipe.text_encoder=CLIPTextModel.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='text_encoder',token=True).to(device=device, dtype=torch.bfloat16)
+    pipe.text_encoder_2=CLIPTextModelWithProjection.from_pretrained('ford442/RealVisXL_V5.0_BF16', subfolder='text_encoder_2',token=True).to(device=device, dtype=torch.bfloat16)
     options = {
         "prompt": [prompt],
         "negative_prompt": [negative_prompt],