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 |
|