Spaces:

stale2000
/

DnDItem

Runtime error

stale2000 commited on Dec 17, 2022

Commit

88c8f2a

1 Parent(s): 8150246

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ from torch import autocast
 openai.api_key = os.getenv('openaikey')
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = StableDiffusionPipeline.from_pretrained("stale2000/sd-dnditem", torch_dtype=torch.float16)
 pipe = pipe.to(device)
 def predict(input, manual_query_repacement, history=[]):

 openai.api_key = os.getenv('openaikey')
 device = "cuda" if torch.cuda.is_available() else "cpu"
+pipe = StableDiffusionPipeline.from_pretrained("stale2000/sd-dnditem", torch_dtype=torch.float16, use_auth_token=True)
 pipe = pipe.to(device)
 def predict(input, manual_query_repacement, history=[]):