Hugging Face's logo

Spaces:
blanchon
/
HiDream-ai-fast
Paused

App Files Community
blanchon commited on
Commit
00e44cf
·
1 Parent(s): 0343e87

up

Browse files
Files changed (1) hide show
  1. app-fast.py +1 -1
app-fast.py CHANGED
@@ -77,7 +77,7 @@ pipe = HiDreamImagePipeline.from_pretrained(
77
  text_encoder_4=text_encoder,
78
  device_map="balanced",
79
  torch_dtype=torch.bfloat16,
80
- ).to(device)
81
 
82
 
83
  @spaces.GPU(duration=120)
 
77
  text_encoder_4=text_encoder,
78
  device_map="balanced",
79
  torch_dtype=torch.bfloat16,
80
+ )
81
 
82
 
83
  @spaces.GPU(duration=120)