rainbow_media_x

Running on Zero

panelforge commited on Sep 18, 2024

Commit

8fe98c0

verified ·

1 Parent(s): 522d992

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ if torch.cuda.is_available():
 else:
     torch_dtype = torch.float32
-pipeline = DiffusionPipeline.from_pretrained("John6666/wai-ani-nsfw-ponyxl-v8-sdxl")
 pipe = pipe.to(device)
 MAX_SEED = np.iinfo(np.int32).max

 else:
     torch_dtype = torch.float32
+pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 MAX_SEED = np.iinfo(np.int32).max