Spaces:

czl
/

generative-data-augmentation-demo

Running on Zero

czl commited on Jun 14

Commit

a6328e5

•

1 Parent(s): 6f4b9a2

update device to cuda only

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ from transformers import CLIPModel, CLIPProcessor
 from tools import synth
-device = "cuda" if torch.cuda.is_available() else "cpu"
 model_path = "runwayml/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 = "runwayml/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")