jadechoghari commited on
Commit
248b236
1 Parent(s): bfffca7

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -6
pipeline.py CHANGED
@@ -7,12 +7,7 @@ from huggingface_hub import hf_hub_download
7
  from safetensors.torch import load_file
8
  import os
9
  from .vae import AutoencoderKL
10
- file_path = hf_hub_download(repo_id="jadechoghari/mar", filename="mar.py")
11
- spec = importlib.util.spec_from_file_location("mar", file_path)
12
- mar = importlib.util.module_from_spec(spec)
13
- sys.modules["mar"] = mar
14
- spec.loader.exec_module(mar)
15
- print("Successfully imported mar module!")
16
 
17
  # inheriting from DiffusionPipeline for HF
18
  class MARModel(DiffusionPipeline):
 
7
  from safetensors.torch import load_file
8
  import os
9
  from .vae import AutoencoderKL
10
+ from .mar import mar
 
 
 
 
 
11
 
12
  # inheriting from DiffusionPipeline for HF
13
  class MARModel(DiffusionPipeline):