lnyan commited on
Commit
0a683e1
·
1 Parent(s): 79bfb65
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def load_t5(device: str | torch.device = "cuda", max_length: int = 512) -> HFEmb
52
  def load_clip(device: str | torch.device = "cuda") -> HFEmbedder:
53
  return HFEmbedder("openai/clip-vit-large-patch14", max_length=77, torch_dtype=torch.bfloat16).to("cuda")
54
 
55
- @spaces.GPU(duration=20)
56
  def load_encoders():
57
  is_schnell = True
58
  t5 = load_t5("cuda", max_length=256 if is_schnell else 512)
 
52
  def load_clip(device: str | torch.device = "cuda") -> HFEmbedder:
53
  return HFEmbedder("openai/clip-vit-large-patch14", max_length=77, torch_dtype=torch.bfloat16).to("cuda")
54
 
55
+ # @spaces.GPU(duration=20)
56
  def load_encoders():
57
  is_schnell = True
58
  t5 = load_t5("cuda", max_length=256 if is_schnell else 512)