Spaces:

rafaaa2105
/

diffusers-collection

Running on Zero

rafaaa2105 commited on Mar 31

Commit

5629139

•

1 Parent(s): 34c1821

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ model_list = os.environ.get("MODELS").split(",")
 lora_list = os.environ.get("LORAS") # Not in use
 @spaces.GPU
-def generate(prompt, model)
     pipe = DiffusionPipeline.from_pretrained(
         model,
         torch_dtype=torch.float16,

 lora_list = os.environ.get("LORAS") # Not in use
 @spaces.GPU
+def generate(prompt, model):
     pipe = DiffusionPipeline.from_pretrained(
         model,
         torch_dtype=torch.float16,