Commit ·
b49212f
1
Parent(s): c8514e2
fix vertexai error
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ from vertexai.generative_models import GenerativeModel, Part,Image
|
|
| 28 |
import vertexai.preview.generative_models as generative_models
|
| 29 |
import time
|
| 30 |
import typing_extensions as typing
|
| 31 |
-
|
| 32 |
torch.set_default_device("cpu")
|
| 33 |
# Initialize the text-to-image pipeline
|
| 34 |
pipe = AutoPipelineForText2Image.from_pretrained(
|
|
@@ -86,7 +86,7 @@ safety_settings = {
|
|
| 86 |
generative_models.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
|
| 87 |
generative_models.HarmCategory.HARM_CATEGORY_HARASSMENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
|
| 88 |
}
|
| 89 |
-
|
| 90 |
|
| 91 |
|
| 92 |
|
|
|
|
| 28 |
import vertexai.preview.generative_models as generative_models
|
| 29 |
import time
|
| 30 |
import typing_extensions as typing
|
| 31 |
+
vertexai.init(project="wubble2024", location="asia-southeast1")
|
| 32 |
torch.set_default_device("cpu")
|
| 33 |
# Initialize the text-to-image pipeline
|
| 34 |
pipe = AutoPipelineForText2Image.from_pretrained(
|
|
|
|
| 86 |
generative_models.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
|
| 87 |
generative_models.HarmCategory.HARM_CATEGORY_HARASSMENT: generative_models.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
|
| 88 |
}
|
| 89 |
+
|
| 90 |
|
| 91 |
|
| 92 |
|