Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -119,7 +119,7 @@ TOKEN = os.getenv('HF_TOKEN')
|
|
| 119 |
MODEL_ID = 'meta-llama/Llama-2-70b-chat-hf'
|
| 120 |
|
| 121 |
pipe = StableDiffusionPipeline.from_pretrained(
|
| 122 |
-
"nota-ai/bk-sdm-
|
| 123 |
)
|
| 124 |
|
| 125 |
STYLES = """
|
|
|
|
| 119 |
MODEL_ID = 'meta-llama/Llama-2-70b-chat-hf'
|
| 120 |
|
| 121 |
pipe = StableDiffusionPipeline.from_pretrained(
|
| 122 |
+
"nota-ai/bk-sdm-base", torch_dtype=torch.float16
|
| 123 |
)
|
| 124 |
|
| 125 |
STYLES = """
|