linoyts HF Staff commited on
Commit
906d441
·
verified ·
1 Parent(s): af3a2cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -5
app.py CHANGED
@@ -24,10 +24,19 @@ subprocess.run(
24
  )
25
 
26
  print("Installing ltx-core and ltx-pipelines from cloned repo...")
 
 
 
 
 
 
27
  subprocess.run(
28
- [sys.executable, "-m", "pip", "install", "--force-reinstall", "--no-deps", "-e",
29
- os.path.join(LTX_REPO_DIR, "packages", "ltx-core"),
30
- "-e", os.path.join(LTX_REPO_DIR, "packages", "ltx-pipelines")],
 
 
 
31
  check=True,
32
  )
33
 
@@ -133,9 +142,11 @@ def build_pipeline(lora_name: str) -> ICLoraPipeline:
133
  distilled_checkpoint_path=checkpoint_path,
134
  spatial_upsampler_path=spatial_upsampler_path,
135
  gemma_root=gemma_root,
136
- loras=[lora],
137
  #loras=[],
138
- quantization=QuantizationPolicy.fp8_cast(),
 
 
139
  )
140
  return pipe
141
 
 
24
  )
25
 
26
  print("Installing ltx-core and ltx-pipelines from cloned repo...")
27
+ # subprocess.run(
28
+ # [sys.executable, "-m", "pip", "install", "--force-reinstall", "--no-deps", "-e",
29
+ # os.path.join(LTX_REPO_DIR, "packages", "ltx-core"),
30
+ # "-e", os.path.join(LTX_REPO_DIR, "packages", "ltx-pipelines")],
31
+ # check=True,
32
+ # )
33
  subprocess.run(
34
+ [
35
+ sys.executable, "-m", "pip", "install",
36
+ "--force-reinstall",
37
+ "-e", os.path.join(LTX_REPO_DIR, "packages", "ltx-core") + "[fp8-trtllm]",
38
+ "-e", os.path.join(LTX_REPO_DIR, "packages", "ltx-pipelines"),
39
+ ],
40
  check=True,
41
  )
42
 
 
142
  distilled_checkpoint_path=checkpoint_path,
143
  spatial_upsampler_path=spatial_upsampler_path,
144
  gemma_root=gemma_root,
145
+ # loras=[lora],
146
  #loras=[],
147
+ # quantization=QuantizationPolicy.fp8_cast(),
148
+ quantization=QuantizationPolicy.fp8_scaled_mm()
149
+
150
  )
151
  return pipe
152