Spaces:

basit123796
/

apnadalle

Sleeping

basit123796 commited on Jan 29, 2024

Commit

af38a18

verified ·

1 Parent(s): 3a0ea23

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -69,7 +69,7 @@ def generate(
     use_resolution_binning: bool = True,
     progress=gr.Progress(track_tqdm=True),
 ):
-    pipe.to(device)
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)

     use_resolution_binning: bool = True,
     progress=gr.Progress(track_tqdm=True),
 ):
+    pipe = some_function_to_create_pipe()
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)