Spaces:

chansung
/

co-write-with-llama2

Runtime error

chansung commited on Aug 13, 2023

Commit

f34c70f

1 Parent(s): 6f97b29

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -119,7 +119,7 @@ TOKEN = os.getenv('HF_TOKEN')
 MODEL_ID = 'meta-llama/Llama-2-70b-chat-hf'
 pipe = StableDiffusionPipeline.from_pretrained(
-  "nota-ai/bk-sdm-small", torch_dtype=torch.float16
 )
 STYLES = """

 MODEL_ID = 'meta-llama/Llama-2-70b-chat-hf'
 pipe = StableDiffusionPipeline.from_pretrained(
+  "nota-ai/bk-sdm-base", torch_dtype=torch.float16
 )
 STYLES = """