Update app.py

app.py

@@ -22,7 +22,6 @@ hf_token = os.environ["HF_TOKEN"]
 
 import argparse
 import os
-#os.environ['CUDA_VISIBLE_DEVICES'] = '7'
 import builtins
 import json
 import math
@@ -51,9 +50,6 @@ from multiprocessing import Process,Queue,set_start_method,get_context
 class ModelFailure:
     pass
 
-gemma_path = "/home/user/app/checkpoints/text_encoder"
-flux_path = "/home/user/app/checkpoints/vae"
-#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
 # Adapted from pipelines.StableDiffusionXLPipeline.encode_prompt
 def encode_prompt(prompt_batch, text_encoder, tokenizer, proportion_empty_prompts, is_train=True):
     captions = []
@@ -73,7 +69,7 @@ def encode_prompt(prompt_batch, text_encoder, tokenizer, proportion_empty_prompt
             max_length=256,
             truncation=True,
             return_tensors="pt",
-        ).cuda()
+        )
 
         print(f"Text Encoder Device: {text_encoder.device}")
         text_input_ids = text_inputs.input_ids.cuda()