Spaces:

declare-lab
/

JAM

Running on Zero

hungchiayu commited on Jan 27

Commit

a073f0a

1 Parent(s): 629c8f6

Update TangoFlux.py

Files changed (1) hide show

TangoFlux.py CHANGED Viewed

@@ -30,7 +30,7 @@ class TangoFluxInference:
         self.vae = AutoencoderOobleck.from_pretrained("stabilityai/stable-audio-open-1.0",subfolder='vae',token=os.environ['HF_TOKEN'])
         paths = snapshot_download(repo_id=name,token=os.environ['HF_TOKEN'])
-        weights = load_file("{}/tangoflux.safetensors".format(paths))
         with open('{}/config.json'.format(paths),'r') as f:
             config = json.load(f)

         self.vae = AutoencoderOobleck.from_pretrained("stabilityai/stable-audio-open-1.0",subfolder='vae',token=os.environ['HF_TOKEN'])
         paths = snapshot_download(repo_id=name,token=os.environ['HF_TOKEN'])
+        weights = load_file("{}/model_1.safetensors".format(paths))
         with open('{}/config.json'.format(paths),'r') as f:
             config = json.load(f)