Spaces:

czl
/

generative-data-augmentation-demo

Running on Zero

czl commited on Oct 2

Commit

cf489b6

•

1 Parent(s): 8a0699c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ from transformers import CLIPModel, CLIPProcessor
 from tools import synth
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_path = "botp/stable-diffusion-v1-5"
 clip_model = CLIPModel.from_pretrained("openai/clip-vit-large-patch14").to(device)
 clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14")

 from tools import synth
 device = "cuda" if torch.cuda.is_available() else "cpu"
+model_path = "czl/stable-diffusion-v1-5"
 clip_model = CLIPModel.from_pretrained("openai/clip-vit-large-patch14").to(device)
 clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14")