picturegen

Running on Zero

m-ric HF staff commited on 7 days ago

Commit

5af8d83

•

1 Parent(s): f184508

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ def infer(
         seed = random.randint(0, MAX_SEED)
     try:
-        generator = torch.Generator(device=device).manual_seed(seed)
         image = pipe(
             prompt=prompt,

         seed = random.randint(0, MAX_SEED)
     try:
+        generator = torch.Generator(device="cuda").manual_seed(seed)
         image = pipe(
             prompt=prompt,