internal_development

Runtime error

Menyu commited on Nov 5, 2024

Commit

9b6ce41

verified ·

1 Parent(s): b802152

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ ENABLE_CPU_OFFLOAD = "1"
 if torch.cuda.is_available():
     pipe = AutoPipelineForText2Image.from_pretrained(
-        "advokat/noobaiXLNAIXL_epsilonPred075",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False

 if torch.cuda.is_available():
     pipe = AutoPipelineForText2Image.from_pretrained(
+        "WhiteAiZ/noobaiXLNAIXL_epsilonPred075_diffusers",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False