Spaces:

huai
/

chinese_stable_diffusion

Runtime error

huai commited on Nov 22, 2022

Commit

97ee091

1 Parent(s): bdfc7c7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from torch import autocast
 from transformers import pipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_id = "CompVis/stable-diffusion-v1-4"
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-zh-en")
 pipe = StableDiffusionPipeline.from_pretrained(
     model_id,

 from transformers import pipeline
 device = "cuda" if torch.cuda.is_available() else "cpu"
+model_id = "stable-diffusion-v1-4"
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-zh-en")
 pipe = StableDiffusionPipeline.from_pretrained(
     model_id,