update

script/clone_model.sh

@@ -0,0 +1,6 @@
+HF_TOKEN="your_token_here"
+huggingface-cli login --token $HF_TOKEN
+MODEL_ID="meta-llama/Llama-3.2-1B-Instruct"
+echo "Downloading ${MODEL_ID}..."
+git lfs install
+git clone "https://huggingface.co/${MODEL_ID}"

script/clone_repo.sh

@@ -6,4 +6,4 @@ git lfs install
 git clone $TARGET_REPO
 cd $REPO_NAME
 source .venv/bin/activate
-uv pip install -e .
+uv pip install -e .

script/upload_model.py

@@ -0,0 +1,29 @@
+import os
+import tarfile
+from huggingface_hub import HfApi
+
+# Initialize Hugging Face API
+api = HfApi()
+# Define paths and filenames
+source_dir = "/workspace/logs/llm4music_diffusion/full_channel_concat"
+output_filename = "full_channel_concat.tar.gz"
+print(f"Creating tar.gz archive from {source_dir}...")
+# Create tar.gz archive
+with tarfile.open(output_filename, "w:gz") as tar:
+    tar.add(source_dir, arcname=os.path.basename(source_dir))
+print("Archive created successfully")
+
+# Upload to Hugging Face
+print("Uploading to Hugging Face Hub...")
+api.upload_file(
+    path_or_fileobj=output_filename,
+    path_in_repo=output_filename,
+    repo_id="seungheondoh/model_temp",  # Replace with your actual repo name
+    repo_type="model"
+)
+
+print("Upload completed successfully")
+
+# Clean up the local tar.gz file
+os.remove(output_filename)
+print("Local archive cleaned up")

script/virtualenv.sh

@@ -40,21 +40,13 @@ else
     echo "Virtual environment already exists."
 fi
 
+# Activate virtual environment
+source venv/bin/activate
+
 # Upgrade pip
 echo "Upgrading pip..."
 pip install --upgrade pip
 
-# Install uv
-echo "Installing uv package manager..."
-curl -LsSf https://astral.sh/uv/install.sh | sh
-
-# Start uv virtualenv
-echo "Starting uv virtualenv..."
-uv venv .venv
-source .venv/bin/activate
-
-# Ensure uv is in PATH
-export PATH="$HOME/.cargo/bin:$PATH"
 # Get CUDA version number if available
 if [ "$CUDA_AVAILABLE" -eq 1 ]; then
     CUDA_VERSION_NUM=$(nvidia-smi --query-gpu=driver_version --format=csv,noheader | head -n 1 | cut -d'.' -f1)
@@ -75,17 +67,25 @@ fi
 
 # Install PyTorch 2.6.0
 echo "Installing PyTorch 2.6.0..."
-echo "Installing PyTorch with CUDA support (${PYTORCH_CUDA})..."
-uv pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/${PYTORCH_CUDA}
+if [ "$CUDA_AVAILABLE" -eq 1 ]; then
+    echo "Installing PyTorch with CUDA support (${PYTORCH_CUDA})..."
+    pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/${PYTORCH_CUDA}
+else
+    echo "Installing CPU-only PyTorch..."
+    pip install torch==2.6.0
+fi
 
 # Install Hugging Face CLI and tools
 echo "Installing Hugging Face CLI and tools..."
-uv pip install huggingface_hub
+pip install huggingface_hub
+
 # Verify installation
 echo "Verifying installation..."
 python -c "import torch; print('PyTorch version:', torch.__version__); print('CUDA available:', torch.cuda.is_available())"
+
 # Check for Hugging Face CLI
 python -c "import huggingface_hub; print('Hugging Face Hub version:', huggingface_hub.__version__)"
+
 echo "============================================================"
 echo "Setup complete!"
 echo "To activate this environment, run:"