Spaces:

prithivMLmods
/

bellatrix-llama-tiny3-1b

Running

prithivMLmods commited on Jun 13, 2024

Commit

b427086

verified ·

1 Parent(s): eb85b5c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,7 +74,7 @@ NUM_IMAGES_PER_PROMPT = 1
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(
-        "Yntec/3Danimation",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False,

 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(
+        "Corcelio/mobius",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False,