Spaces:

machinesarenotpeople
/

ruinsoftomorrow

Runtime error

machinesarenotpeople commited on Dec 25, 2022

Commit

23cd458

•

1 Parent(s): e4bc1c3

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ USER_TOKEN=os.environ.get('HF_TOKEN_SD')
 #1.5
 #model_id = "runwayml/stable-diffusion-v1-5"
 #pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16", use_auth_token = USER_TOKEN)
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype = "auto", use_auth_token = USER_TOKEN)
 pipe.to(device)

 #1.5
 #model_id = "runwayml/stable-diffusion-v1-5"
 #pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16, revision="fp16", use_auth_token = USER_TOKEN)
+pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token = USER_TOKEN)
 pipe.to(device)