atef91 commited on
Commit
4c69a95
·
verified ·
1 Parent(s): 08c73ce

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import warnings
6
  warnings.filterwarnings('ignore')
7
 
8
  # Constants
9
- MODEL_ID = "runwayml/stable-diffusion-v1-5"
10
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
11
  DTYPE = torch.float32 # Using float32 for better compatibility
12
 
@@ -21,7 +21,7 @@ class FooocusGenerator:
21
  self.pipe = StableDiffusionPipeline.from_pretrained(
22
  MODEL_ID,
23
  torch_dtype=DTYPE,
24
- use_auth_token=True # For authenticated access to model
25
  )
26
 
27
  if DEVICE == "cuda":
 
6
  warnings.filterwarnings('ignore')
7
 
8
  # Constants
9
+ MODEL_ID = "CompVis/stable-diffusion-v1-4" # Using v1.4 which is publicly available
10
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
11
  DTYPE = torch.float32 # Using float32 for better compatibility
12
 
 
21
  self.pipe = StableDiffusionPipeline.from_pretrained(
22
  MODEL_ID,
23
  torch_dtype=DTYPE,
24
+ use_auth_token=False # No auth token needed for v1.4
25
  )
26
 
27
  if DEVICE == "cuda":