MenYuImg

Running on Zero

Menyu commited on Nov 5, 2024

Commit

368c509

verified ·

1 Parent(s): 03b5f6a

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(
-        "John6666/noobai-xl-nai-xl-epsilonpred075version-sdxl",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False

 if torch.cuda.is_available():
     pipe = AutoPipelineForText2Image.from_pretrained(
+        "https://huggingface.co/advokat/noobaiXLNAIXL_epsilonPred075/blob/main/noobaiXLNAIXL_epsilonPred075.safetensors",
         torch_dtype=torch.float16,
         use_safetensors=True,
         add_watermarker=False