Hugging Face's logo

Spaces:
ai-forever
/
kandinsky-4-t2v-flash
Running on Zero

App Files Community
ai-forever commited on
Commit
c369c40
•
1 Parent(s): cc8e199

uncomment pipe

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,4 +1,3 @@
1
- # исправленная версия (чтобы не потерялась)
2
  import subprocess
3
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
4
  # subprocess.run('pip install bitsandbytes', shell=True)
@@ -38,7 +37,7 @@ device_map = {
38
  "vae": torch.device('cuda'),
39
  "text_embedder": torch.device('cuda')
40
  }
41
- # pipe = get_T2V_pipeline(device_map)
42
 
43
  max_64_bit_int = 2**63 - 1
44
 
 
 
1
  import subprocess
2
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
3
  # subprocess.run('pip install bitsandbytes', shell=True)
 
37
  "vae": torch.device('cuda'),
38
  "text_embedder": torch.device('cuda')
39
  }
40
+ pipe = get_T2V_pipeline(device_map)
41
 
42
  max_64_bit_int = 2**63 - 1
43