HiDream-ai-fast

Paused

blanchon commited on Apr 7

Commit

00e44cf

1 Parent(s): 0343e87

up

Files changed (1) hide show

app-fast.py CHANGED Viewed

@@ -77,7 +77,7 @@ pipe = HiDreamImagePipeline.from_pretrained(
     text_encoder_4=text_encoder,
     device_map="balanced",
     torch_dtype=torch.bfloat16,
-).to(device)
 @spaces.GPU(duration=120)

     text_encoder_4=text_encoder,
     device_map="balanced",
     torch_dtype=torch.bfloat16,
+)
 @spaces.GPU(duration=120)