Muhammad Anas Akhtar
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,6 @@ def image_generation(prompt):
|
|
15 |
# Load the Stable Diffusion 3 pipeline
|
16 |
pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1",
|
17 |
torch_dtype=torch.float16 if device == "cuda" else torch.float32,
|
18 |
-
use_auth_token=HUGGINGFACE_TOKEN, # Use the Hugging Face token for authentication
|
19 |
text_encoder_3=None,
|
20 |
tokenizer_3=None)
|
21 |
|
|
|
15 |
# Load the Stable Diffusion 3 pipeline
|
16 |
pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1",
|
17 |
torch_dtype=torch.float16 if device == "cuda" else torch.float32,
|
|
|
18 |
text_encoder_3=None,
|
19 |
tokenizer_3=None)
|
20 |
|