Spaces:

philipp-zettl
/

t2i-model-2

Build error

philipp-zettl commited on May 30, 2024

Commit

8f007ec

verified ·

1 Parent(s): a3d77a0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,6 @@ model_name = os.environ.get('MODEL_REPO_ID')
 pipe = DiffusionPipeline.from_pretrained(
     model_name,
     torch_dtype=torch.float16,
-    variant="fp16"
 )
 def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width, height, num_samples):

 pipe = DiffusionPipeline.from_pretrained(
     model_name,
     torch_dtype=torch.float16,
 )
 def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width, height, num_samples):