Spaces:

multimodalart
/

AuraFlow

Running on Zero

multimodalart HF Staff commited on Jul 12, 2024

Commit

b39d4ca

verified ·

1 Parent(s): 93fd450

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 #torch._inductor.config.coordinate_descent_check_all_directions = True
 pipe = AuraFlowPipeline.from_pretrained(
-	"AuraDiffusion/AuraFlow",
     torch_dtype=torch.float16
 ).to("cuda")

 #torch._inductor.config.coordinate_descent_check_all_directions = True
 pipe = AuraFlowPipeline.from_pretrained(
+	"fal/AuraFlow",
     torch_dtype=torch.float16
 ).to("cuda")