Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,7 +64,8 @@ models = []
|
|
| 64 |
|
| 65 |
my_token = os.environ['api_key']
|
| 66 |
|
| 67 |
-
pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2", revision="fp16", torch_dtype=torch.float16, use_auth_token=my_token).to("cuda")
|
|
|
|
| 68 |
|
| 69 |
|
| 70 |
def check_prompt(prompt):
|
|
|
|
| 64 |
|
| 65 |
my_token = os.environ['api_key']
|
| 66 |
|
| 67 |
+
# pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2", revision="fp16", torch_dtype=torch.float16, use_auth_token=my_token).to("cuda")
|
| 68 |
+
pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2", variant="fp16", torch_dtype=torch.float16, use_auth_token=my_token).to("cuda")
|
| 69 |
|
| 70 |
|
| 71 |
def check_prompt(prompt):
|