Upload 3 files

app.py

@@ -8,7 +8,7 @@ warnings.filterwarnings('ignore')
 # Constants
 MODEL_ID = "runwayml/stable-diffusion-v1-5"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
+DTYPE = torch.float32  # Using float32 for better compatibility
 
 class FooocusGenerator:
     def __init__(self):
@@ -21,7 +21,7 @@ class FooocusGenerator:
             self.pipe = StableDiffusionPipeline.from_pretrained(
                 MODEL_ID,
                 torch_dtype=DTYPE,
-                revision="fp16" if DEVICE == "cuda" else "fp32"
+                use_auth_token=True  # For authenticated access to model
             )
             
             if DEVICE == "cuda":

requirements.txt

@@ -10,5 +10,3 @@ einops==0.6.1
 pytorch-lightning==1.9.0
 omegaconf==2.3.0
 huggingface-hub>=0.19
-xformers==0.0.13
-triton==2.0.0